fix(trainer_callback]: typo (#8121)

2020-10-28 11:15:30 -05:00
parent 5423f2a9d4
commit b4cacb7a63
1 changed files with 1 additions and 1 deletions
--- a/src/transformers/trainer_callback.py
+++ b/src/transformers/trainer_callback.py
@@ -148,7 +148,7 @@ class TrainerControl:

    def _new_step(self):
        """ Internal method that resets the variable for a new step. """
-        self.should_save_model = False
+        self.should_save = False
        self.should_evaluate = False
        self.should_log = False