[EncoderDecoderConfig] automatically set decoder config to decoder (#4809)

* automatically set decoder config to decoder

* add more tests
This commit is contained in:
Patrick von Platen
2020-06-05 23:16:37 +02:00
committed by GitHub
parent f1fe18465d
commit 8cca875569
2 changed files with 40 additions and 1 deletions

View File

@@ -85,6 +85,9 @@ class EncoderDecoderConfig(PretrainedConfig):
Returns:
:class:`EncoderDecoderConfig`: An instance of a configuration object
"""
logger.info("Set `config.is_decoder=True` for decoder_config")
decoder_config.is_decoder = True
return cls(encoder=encoder_config.to_dict(), decoder=decoder_config.to_dict())
def to_dict(self):