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

edbbc0f2 · Jintao Lin · GitHub · 6b43ae37 · edbbc0f2
Unverified Commit edbbc0f2 authored Aug 06, 2020 by Jintao Lin Committed by GitHub Aug 06, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

mmcv/runner/hooks/lr_updater.py mmcv/runner/hooks/lr_updater.py +2 -2

No files found.
--- a/mmcv/runner/hooks/lr_updater.py
+++ b/mmcv/runner/hooks/lr_updater.py
@@ -110,11 +110,11 @@ class LrUpdaterHook(Hook):
                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)
            self.warmup_iters = self.warmup_epochs * epoch_len

-    def before_train_epoch(self, runner):
        if not self.by_epoch:
            return