saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
transformers/tests/optimization
History
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
..
__init__.py
test_optimization.py Make schedulers picklable by making lr_lambda fns global (#21768) 2023-03-02 12:08:43 -05:00
test_optimization_tf.py