diff --git a/src/transformers/trainer_callback.py b/src/transformers/trainer_callback.py index 398d5fd3513e..01be518da1d0 100644 --- a/src/transformers/trainer_callback.py +++ b/src/transformers/trainer_callback.py @@ -148,7 +148,7 @@ def _new_epoch(self): 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