7 Commits

Author	SHA1	Message	Date
RangiLyu	813f49bf23	[Feature] Support CosineRestartParamScheduler. (#397) ... * [Feature] Support CosineRestartParamScheduler. * add ut and docstring * add docstring	2022-08-11 17:57:35 +08:00
Ma Zerun	3da66d1f87	[Enhance] Auto set the end of param schedulers. (#361) ... * [Enhance] Auto set the `end` of param schedulers. * Add log output and unit test * Update docstring * Update unit tests of `CosineAnnealingParamScheduler`.	2022-07-15 19:53:28 +08:00
RangiLyu	a3d2916790	[Enhance] Support scheduling betas with MomentumScheduler. (#346) ... * [Enhance] Support scheduling betas with MomentumScheduler. * enhance ut * test adam betas * enhance ut * enhance ut	2022-07-05 20:37:23 +08:00
Tong Gao	c3aff4fc9a	[Enhancement] Add PolyParamScheduler, PolyMomentum and PolyLR (#188) ... * [Enhancement] Add PolyParamScheduler, PolyMomentum and PolyLR * min_lr -> eta_min, refined docstr	2022-04-25 13:44:15 +08:00
RangiLyu	c2c5664fad	Fix pt1.5 unit tests. (#65) ... * Fix pt1.5 unit tests. * move to mmengine.testing	2022-03-01 11:28:21 +08:00
RangiLyu	7905f039b6	[Feature]: Add parameter schedulers. (#22) ... * [Feature]: Add parameter schedulers. * update * update * update * update * add docstring to lr and momentum * resolve comments	2022-02-16 23:43:59 +08:00
RangiLyu	bbb7d625e6	add scheduler unit test (#13) ... * tmp * add scheduler unit test * disable yapf * add more test * add more test * not use torch test case * solve comments * update file * add more unit tests * resolve comments * update cosine ut * fix typo * solve comments * solve comments * resolve comments	2022-02-16 22:33:46 +08:00