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 refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_optimization_tf.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_optimization.py

Make schedulers picklable by making lr_lambda fns global (#21768)

2023-03-02 12:08:43 -05:00