From 4002cb85b5f4bd84ceb445867c9b287d6edeed1e Mon Sep 17 00:00:00 2001 From: Zach Mueller Date: Mon, 29 Apr 2024 14:26:47 -0400 Subject: [PATCH] Update comment based on sourab feedback --- 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 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