Commit 09379ad8 authored by alexeib's avatar alexeib Committed by Myle Ott
Browse files

add big en_fr transformer architecture

parent e774fda7

fairseq/models/transformer.py

+5 −0
Changes for fairseq/models/transformer.py: 5 added lines, 0 removed lines.
Original line number Diff line number Diff line
@@ -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):