Skip to content

GitLab

Unverified Commit d6210e6d authored by Wenwei Zhang's avatar Wenwei Zhang Committed by GitHub
Browse files

Fix cyclic registry (#249) 

* track progress of iter&enum

* restore

* Add CyclicLrUpdater in Hooks
parent a338d43d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
mmcv/runner/hooks/lr_updater.py
View file @ d6210e6d
...@@ -221,6 +221,7 @@ class CosineAnealingLrUpdaterHook(LrUpdaterHook): ...@@ -221,6 +221,7 @@ class CosineAnealingLrUpdaterHook(LrUpdaterHook):
return annealing_cos(base_lr, target_lr, progress / max_progress) return annealing_cos(base_lr, target_lr, progress / max_progress)
@HOOKS.register_module
class CyclicLrUpdaterHook(LrUpdaterHook): class CyclicLrUpdaterHook(LrUpdaterHook):
"""Cyclic LR Scheduler """Cyclic LR Scheduler
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment