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.py	Make schedulers picklable by making lr_lambda fns global (#21768)	2023-03-02 12:08:43 -05:00
test_optimization_tf.py	[Test refactor 1/5] Per-folder tests reorganization (#15725)	2022-02-23 15:46:28 -05:00