Commit 316744d6 authored by Angela Fan's avatar Angela Fan Committed by Myle Ott
Browse files

minor parameter fixes for stories model

parent 7a36da42

fairseq/models/fconv_self_att.py

+0 −6
Changes for fairseq/models/fconv_self_att.py: 0 added lines, 6 removed lines.
Original line number Diff line number Diff line
@@ -499,12 +499,6 @@ def fconv_self_att_wp(args): 
    args.decoder_embed_dim = getattr(args, 'decoder_embed_dim', 256)

    args.decoder_layers = getattr(args, 'decoder_layers', '[(512, 4)] * 4 + [(768, 4)] * 2 + [(1024, 4)] * 1')

    args.decoder_out_embed_dim = getattr(args, 'decoder_out_embed_dim', 256)

    args.decoder_attention = getattr(args, 'decoder_attention', 'True')

    args.multihead_attention_nheads = getattr(args, 'multihead_attention_nheads', 1)

    args.encoder_attention = getattr(args, 'encoder_attention', 'False')

    args.encoder_attention_nheads = getattr(args, 'encoder_attention_nheads', 1)

    args.self_attention = getattr(args, 'self_attention', 'True')

    args.gated_attention = getattr(args, 'gated_attention', 'True')

    args.downsample = getattr(args, 'downsample', 'True')

    args.project_input = getattr(args, 'project_input', 'True')

    args.multihead_self_attention_nheads = getattr(args, 'multihead_self_attention_nheads', 4)