Merge pull request #465 from QData/run-training-bug

Run training bug
QData · May 23, 2021 · 6950621 · 6950621
2 parents e9fde99 + facad24
commit 6950621
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/textattack/commands/train_model/run_training.py b/textattack/commands/train_model/run_training.py
@@ -552,7 +552,7 @@ def loss_backward(loss):
             tb_writer.add_scalar("epoch_eval_score", eval_score, epoch)
 
             if args.checkpoint_every_epoch:
-                _save_model_checkpoint(model, args.output_dir, args.global_step)
+                _save_model_checkpoint(model, args.output_dir, global_step)
 
             logger.info(
                 f"Eval {'pearson correlation' if args.do_regression else 'accuracy'}: {eval_score*100}%"