Yih-Dar
c600e89f5c
Update unwrap_and_save_reload_schedule to use weights_only=False ( #35952 )
...
* fix
* Fix
---------
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2025-01-29 14:30:57 +01:00
Sai-Suraj-27
2e48b3e872
fix: Fixed the 1st argument name in classmethods ( #31907 )
...
Fixed the first argument name in few classmethods.
2024-07-11 12:11:50 +01:00
zspo
cda9c82a63
fix get_scheduler when name is warmup_stable_decay ( #31128 )
...
fix get_scheduler args
2024-05-30 15:25:43 +01:00
Alexander Visheratin
7b1170b0fa
Add WSD scheduler ( #30231 )
...
* Added WSD scheduler.
* Added tests.
* Fixed errors.
* Fix formatting.
* CI fixes.
2024-04-25 12:07:21 +01:00
Connor Henderson
8e5a1b2abb
Make schedulers picklable by making lr_lambda fns global ( #21768 )
...
* Make schedulers picklable by making lr_lambda fns global
* add unused _get_constant_schedule_lr_lambda arg
* remove unneeded _get_constant_schedule_lr_lamda
* add test
* make style
* rebase, remove torch dep, put lambda back
* repo-consistency and style
2023-03-02 12:08:43 -05:00
Adrian Sager La Ganga
a3034c7004
Add inverse sqrt learning rate scheduler ( #21495 )
...
* added inverse sqrt lr scheduler
* Updated get_scheduler in src/transformers/optimization.py
* Updated src/transformers/__init__.py
* Added inverse sqrt lr scheduler test
* Updated docs/source/en/main_classes/optimizer_schedules.mdx
* Ran style and quality scripts
* Fix get_inverse_sqrt_schedule docstring
* Comment implementation URL
2023-02-07 15:00:50 -05:00
Sylvain Gugger
6f79d26442
Update quality tooling for formatting ( #21480 )
...
* Result of black 23.1
* Update target to Python 3.7
* Switch flake8 to ruff
* Configure isort
* Configure isort
* Apply isort with line limit
* Put the right black version
* adapt black in check copies
* Fix copies
2023-02-06 18:10:56 -05: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
