David Hou 9f66dcf718 PolynomialDecayWithWarmup + tests (#3649) ... * working PolynomialDecayWithWarmup + tests....... add lars_util.py, oops * keep lars_util.py as intact as possible, simplify our interface * whitespace * clean up * clean up * asserts * test polylr for full resnet training run * add comment * rename * fix do_optim * don't cast lr * info * calculate from train_files * skip it		2024-03-07 18:53:36 -05:00
..
lars_optimizer.py	PolynomialDecayWithWarmup + tests (#3649)	2024-03-07 18:53:36 -05:00
lars_util.py	PolynomialDecayWithWarmup + tests (#3649)	2024-03-07 18:53:36 -05:00