10 Commits

Author	SHA1	Message	Date
Nosimus	effaef334b	fix: prevent second save in the end of training if last step was saved already (#36219) ... * fix: prevent second save in the end of training * fix: prevent second save in the end of training * test: added test for no duplicate save on epoch save strategy * fix: removed TrainerControl * chore: style formatting --------- Co-authored-by: JaktensTid <jaktenstid1@gmail.com>	2025-02-20 17:38:52 +01:00
Wing Lian	1027a532c5	add a callback hook right before the optimizer step (#33444)	2024-09-13 10:43:45 +02:00
Yih-Dar	080e14b24c	Modify warnings in a with block to avoid flaky tests (#31893) ... * fix * [test_all] check before merge --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>	2024-07-10 17:56:12 +02:00
Dhruv Pai	5c88253556	Add on_optimizer_step to callback options (#31095) ... * Modified test * Added on_optimizer_step to callbacks * Move callback after step is called * Added on optimizer step callback	2024-05-29 16:20:59 +02:00
Zach Mueller	daf281f44f	Enforce saving at end of training if saving option chosen (#30160) ... * Enforce saving at end of training * Fix test * Rework test * Fixup tests' * Update comment based on sourab feedback * Clean	2024-05-21 07:50:11 -04:00
Zach Mueller	ad697f1801	Introduce Stateful Callbacks (#29666) ... * Introduce saveable callbacks * Add note * Test for non-present and flag * Support early stopping and refusing to train further * Update docstring * More saving * Import oopsie * Apply suggestions from code review Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> * Make it go through TrainerArguments * Document * Fix test * Apply suggestions from code review Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> * Rework to allow for duplicates * CLean * Fix failing tests --------- Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>	2024-04-25 11:00:09 -04:00
Zach Mueller	60d5f8f9f0	🚨🚨🚨Deprecate evaluation_strategy to eval_strategy🚨🚨🚨 (#30190) ... * Alias * Note alias * Tests and src * Rest * Clean * Change typing? * Fix tests * Deprecation versions	2024-04-18 12:49:43 -04:00
Aaron Gokaslan	5e8c8eb5ba	Apply ruff flake8-comprehensions (#21694)	2023-02-22 09:14:54 +01:00
neverix	8b332a6a16	Make predict() close progress bars after finishing (#17952) (#18078) ... * Make Trainer.predict call on_evaluate (#17952) * Add on_predict * Small fix * Small and different fix * Add tests	2022-07-08 16:44:24 -04:00
Lysandre Debut	29c10a41d0	[Test refactor 1/5] Per-folder tests reorganization (#15725) ... * Per-folder tests reorganization Co-authored-by: sgugger <sylvain.gugger@gmail.com> Co-authored-by: Stas Bekman <stas@stason.org>	2022-02-23 15:46:28 -05:00