add big en_fr transformer architecture
@@ -432,6 +432,11 @@ def transformer_vaswani_wmt_en_de_big(args):
args.dropout = getattr(args, 'dropout', 0.3)
base_architecture(args)
+@register_model_architecture('transformer', 'transformer_vaswani_wmt_en_fr_big')
+def transformer_vaswani_wmt_en_fr_big(args):
+ args.dropout = getattr(args, 'dropout', 0.1)
+ transformer_vaswani_wmt_en_de_big(args)
+
@register_model_architecture('transformer', 'transformer_wmt_en_de_big')
def transformer_wmt_en_de_big(args):
