Commit d4816034 authored by Myle Ott's avatar Myle Ott
Browse files

Fix old model checkpoints

parent 81e99d8d

fairseq/models/fconv.py

+4 −0
Changes for fairseq/models/fconv.py: 4 added lines, 0 removed lines.
Original line number Diff line number Diff line
@@ -60,6 +60,10 @@ class FConvModel(FairseqModel): 
            args.max_target_positions = args.max_positions

        if not hasattr(args, 'share_input_output_embed'):

            args.share_input_output_embed = False

        if not hasattr(args, 'encoder_embed_path'):

            args.encoder_embed_path = None

        if not hasattr(args, 'decoder_embed_path'):

            args.decoder_embed_path = None



        encoder_embed_dict = None

        if args.encoder_embed_path:

fairseq/models/lstm.py

+4 −0
Changes for fairseq/models/lstm.py: 4 added lines, 0 removed lines.
Original line number Diff line number Diff line
@@ -61,6 +61,10 @@ class LSTMModel(FairseqModel): 
        base_architecture(args)



        """Build a new model instance."""

        if not hasattr(args, 'encoder_embed_path'):

            args.encoder_embed_path = None

        if not hasattr(args, 'decoder_embed_path'):

            args.decoder_embed_path = None



        encoder_embed_dict = None

        if args.encoder_embed_path: