fix(trainer_callback]: typo (#8121)

b4cacb7a · Boris Dayma · GitHub · 5423f2a9 · b4cacb7a
Unverified Commit b4cacb7a authored Oct 28, 2020 by Boris Dayma Committed by GitHub Oct 28, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/trainer_callback.py src/transformers/trainer_callback.py +1 -1

No files found.
--- 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