Skip to content

GitLab

Unverified Commit 2024faf1 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Fix save when laod_best_model_at_end=True (#14054)

parent 2c60ff2f
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 2 deletions
src/transformers/trainer_callback.py
View file @ 2024faf1
...@@ -418,8 +418,6 @@ class DefaultFlowCallback(TrainerCallback): ...@@ -418,8 +418,6 @@ class DefaultFlowCallback(TrainerCallback):
# Evaluate # Evaluate
if args.evaluation_strategy == IntervalStrategy.STEPS and state.global_step % args.eval_steps == 0: if args.evaluation_strategy == IntervalStrategy.STEPS and state.global_step % args.eval_steps == 0:
control.should_evaluate = True control.should_evaluate = True
if args.load_best_model_at_end:
control.should_save = True
# Save # Save
if ( if (
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment