Skip to content

GitLab

Unverified Commit edbbc0f2 authored by Jintao Lin's avatar Jintao Lin Committed by GitHub
Browse files

fix bug for self.warmup_iters when warmup_by_epoch (#470)

parent 6b43ae37
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
mmcv/runner/hooks/lr_updater.py
View file @ edbbc0f2
...@@ -110,11 +110,11 @@ class LrUpdaterHook(Hook): ...@@ -110,11 +110,11 @@ class LrUpdaterHook(Hook):
group['initial_lr'] for group in runner.optimizer.param_groups group['initial_lr'] for group in runner.optimizer.param_groups
] ]
if self.warmup_by_epoch: def before_train_epoch(self, runner):
if self.warmup_iters is None:
epoch_len = len(runner.data_loader) epoch_len = len(runner.data_loader)
self.warmup_iters = self.warmup_epochs * epoch_len self.warmup_iters = self.warmup_epochs * epoch_len
def before_train_epoch(self, runner):
if not self.by_epoch: if not self.by_epoch:
return return
......
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