Skip to content

GitLab

Unverified Commit 5f611bbb authored by lizz's avatar lizz Committed by GitHub
Browse files

Update lr_updater.py (#334)

parent 6988de82
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
mmcv/runner/hooks/lr_updater.py
View file @ 5f611bbb
...@@ -25,8 +25,7 @@ class LrUpdaterHook(Hook): ...@@ -25,8 +25,7 @@ class LrUpdaterHook(Hook):
warmup=None, warmup=None,
warmup_iters=0, warmup_iters=0,
warmup_ratio=0.1, warmup_ratio=0.1,
warmup_by_epoch=False, warmup_by_epoch=False):
**kwargs):
# validate the "warmup" argument # validate the "warmup" argument
if warmup is not None: if warmup is not None:
if warmup not in ['constant', 'linear', 'exp']: if warmup not in ['constant', 'linear', 'exp']:
......
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