fix bugs

fairseq/utils.py

@@ -188,7 +188,7 @@ def load_ensemble_for_inference(filenames, task, model_arg_overrides=None):
     for index,state in enumerate(states):
         model = task.build_model(args[index])
         model.upgrade_state_dict(state['model'])
-        model.load_state_dict(state['model'], strict=False)
+        model.load_state_dict(state['model'], strict=True)
         ensemble.append(model)
     return ensemble, args
 

requirements.txt

 cffi
 numpy
-torch
 tqdm
 sacrebleu
 sacremoses
-tensorboardX
-tensorflow
\ No newline at end of file
+tensorboardX
\ No newline at end of file

train.py

@@ -54,7 +54,6 @@ def main(args):
     model = task.build_model(args)
     # for k,v in model.named_parameters():
     #     print("k:%s"%k)
-    print(model)
     criterion = task.build_criterion(args)
     print('| model {}, criterion {}'.format(args.arch, criterion.__class__.__name__))
     print('| num. model params: {}'.format(sum(p.numel() for p in model.parameters())))

translate.sh

@@ -3,7 +3,7 @@ set -e
 
 model_root_dir=checkpoints
 # set tag
-model_dir_tag=baseline_v2
+model_dir_tag=baseline
 model_dir=$model_root_dir/$model_dir_tag
 
 ensemble=