Merge branch 'master' of https://github.com/ananyahjha93/pytorch-pretrained-BERT

pull current master to local
2019-03-17 08:30:13 -04:00
parent 8a4e90ff40 f3e5404880
commit e5b63fb542
6 changed files with 13 additions and 6 deletions
--- a/examples/run_classifier.py
+++ b/examples/run_classifier.py
@@ -857,7 +857,6 @@ def main():
                    optimizer.zero_grad()
                    global_step += 1

-    if args.do_train:
        # Save a trained model and the associated configuration
        model_to_save = model.module if hasattr(model, 'module') else model  # Only save the model it-self
        output_model_file = os.path.join(args.output_dir, WEIGHTS_NAME)