From b4cacb7a63473b527e3c723576bba81611bc3b16 Mon Sep 17 00:00:00 2001 From: Boris Dayma Date: Wed, 28 Oct 2020 11:15:30 -0500 Subject: [PATCH] fix(trainer_callback]: typo (#8121) --- src/transformers/trainer_callback.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/trainer_callback.py b/src/transformers/trainer_callback.py index 398d5fd351..01be518da1 100644 --- 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