Skip to content

GitLab

Commit 50931d69 authored by alexeib's avatar alexeib Committed by Myle Ott
Browse files

fix default params

parent 7f5f2461
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
fairseq/models/transformer.py
View file @ 50931d69
...@@ -400,8 +400,8 @@ def base_architecture(args): ...@@ -400,8 +400,8 @@ def base_architecture(args):
args.decoder_layers = getattr(args, 'decoder_layers', 6) args.decoder_layers = getattr(args, 'decoder_layers', 6)
args.decoder_attention_heads = getattr(args, 'decoder_attention_heads', 8) args.decoder_attention_heads = getattr(args, 'decoder_attention_heads', 8)
args.attention_dropout = getattr(args, 'attention_dropout', 0.) args.attention_dropout = getattr(args, 'attention_dropout', 0.)
args.attention_dropout = getattr(args, 'relu_dropout', 0.) args.relu_dropout = getattr(args, 'relu_dropout', 0.)
args.attention_dropout = getattr(args, 'dropout', 0.1) args.dropout = getattr(args, 'dropout', 0.1)
@register_model_architecture('transformer', 'transformer_iwslt_de_en') @register_model_architecture('transformer', 'transformer_iwslt_de_en')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment