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