Update comment based on sourab feedback

huggingface · May 15, 2024 · 4002cb8 · 4002cb8
1 parent 370e6cf
commit 4002cb8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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