mmengine

Commit Graph

Author	SHA1	Message	Date
Z-Fran	090104df21	[Fix] Fix the calculation error of eta_min in CosineRestart (#639 ) * [Fix] fix CosineRestart eta_min * add ut case * Enhance unit test Enhance unit test * remove unused code Co-authored-by: HAOCHENYE <21724054@zju.edu.cn>	2022-11-01 15:48:39 +08:00
RangiLyu	813f49bf23	[Feature] Support CosineRestartParamScheduler. (#397 ) * [Feature] Support CosineRestartParamScheduler. * add ut and docstring * add docstring	2022-08-11 17:57:35 +08:00
Mashiro	6b47035fdf	[Fix] Fix save scheduler state dict with optim wrapper (#375 ) * fix save scheduler state dict with optim wrapper * remove for loop and inherit TestParameterScheduler * remove for loop and inherit TestParameterScheduler * minor refine	2022-07-20 16:32:48 +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
Miao Zheng	fd295741ca	[Features]Add OneCycleLR (#296 ) * [Features]Add OnecycleLR * [Features]Add OnecycleLR * yapf disable * build_iter_from_epoch * add epoch * fix args * fix according to comments; * lr-param * fix according to comments * defaults -> default to * remove epoch and steps per step * variabel names	2022-06-13 21:23:59 +08:00
RangiLyu	1912660db9	[Feature] Support convert epoch-based schedulers to iter-based. (#221 ) * [Feature] Support convert epoch-based schedulers to iter-based. * Support convert and refactor LR and Momentum to mixin. * Add unit tests * fix args and add runner ut * resolve comments	2022-05-10 15:17:51 +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

11 Commits (6636f07cfe7c24507d59d94996559ed5273ae115)