Fix cyclic registry (#249)

* track progress of iter&enum * restore * Add CyclicLrUpdater in Hooks

Fix cyclic registry (#249)
* track progress of iter&enum * restore * Add CyclicLrUpdater in Hooks
d6210e6d · Wenwei Zhang · GitHub · a338d43d · d6210e6d
Unverified Commit d6210e6d authored Apr 27, 2020 by Wenwei Zhang Committed by GitHub Apr 27, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

mmcv/runner/hooks/lr_updater.py mmcv/runner/hooks/lr_updater.py +1 -0

No files found.
--- a/mmcv/runner/hooks/lr_updater.py
+++ b/mmcv/runner/hooks/lr_updater.py
@@ -221,6 +221,7 @@ class CosineAnealingLrUpdaterHook(LrUpdaterHook):
        return annealing_cos(base_lr, target_lr, progress / max_progress)


+@HOOKS.register_module
 class CyclicLrUpdaterHook(LrUpdaterHook):
    """Cyclic LR Scheduler