SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Enforce saving at end of training if saving option chosen (#30160)

Browse Source 
* Enforce saving at end of training

* Fix test

* Rework test

* Fixup tests'

* Update comment based on sourab feedback

* Clean
This commit is contained in:
Zach Mueller
2024-05-21 07:50:11 -04:00
committed by GitHub
parent 7a4792e6b3
commit daf281f44f
4 changed files with 58 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/trainer/test_trainer_callback.py
View File

@@ -153,7 +153,7 @@ class TrainerCallbackTest(unittest.TestCase):
expected_events.append("on_log")
if trainer.args.eval_strategy == IntervalStrategy.STEPS and step % trainer.args.eval_steps == 0:
expected_events += evaluation_events.copy()
if step % trainer.args.save_steps == 0:
if step % trainer.args.save_steps == 0 or step == trainer.state.max_steps:
expected_events.append("on_save")
expected_events.append("on_epoch_end")
if trainer.args.eval_strategy == IntervalStrategy.EPOCH: