diff --git a/src/transformers/trainer_callback.py b/src/transformers/trainer_callback.py index 5d005384de41d2..d2570ed8ba4317 100644 --- a/src/transformers/trainer_callback.py +++ b/src/transformers/trainer_callback.py @@ -544,7 +544,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