riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-03 03:31:05 +06:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Fix save when laod_best_model_at_end=True (#14054)

Browse Source
This commit is contained in:
Sylvain Gugger 2021-10-18 10:22:57 -04:00 committed by GitHub
parent 2c60ff2fe2
commit 2024faf171
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/trainer_callback.py
View File

@ -418,8 +418,6 @@ class DefaultFlowCallback(TrainerCallback):
# Evaluate
if args.evaluation_strategy == IntervalStrategy.STEPS and state.global_step % args.eval_steps == 0:
control.should_evaluate = True
if args.load_best_model_at_end:
control.should_save = True
# Save
if (