From cc684e5142dc2f0035aafb389edfc1de7809bcd1 Mon Sep 17 00:00:00 2001 From: Zach Mueller Date: Mon, 29 Apr 2024 13:29:35 -0400 Subject: [PATCH] Update src/transformers/trainer_callback.py Co-authored-by: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com> --- 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 8e7b82e43dd811..f0ad8e56c7d15c 100644 --- a/src/transformers/trainer_callback.py +++ b/src/transformers/trainer_callback.py @@ -460,7 +460,7 @@ def on_step_end(self, args: TrainingArguments, state: TrainerState, control: Tra # End training if state.global_step >= state.max_steps: control.should_training_stop = True - # Save the best model at the end if we have a save strategy + # Save the model at the end if we have a save strategy if args.save_strategy != IntervalStrategy.NO: control.should_save = True