"git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "a0939977a3b3c34c925c565c3fd3dcbe5d09e23c"
Unverified Commit 71b59a8d authored by Jintao Lin's avatar Jintao Lin Committed by GitHub
Browse files

assign `self.warmup_iters` in `before_run` function (#458)

parent 9de04d22
...@@ -110,13 +110,14 @@ class LrUpdaterHook(Hook): ...@@ -110,13 +110,14 @@ class LrUpdaterHook(Hook):
group['initial_lr'] for group in runner.optimizer.param_groups group['initial_lr'] for group in runner.optimizer.param_groups
] ]
def before_train_epoch(self, runner):
if not self.by_epoch:
return
if self.warmup_by_epoch: if self.warmup_by_epoch:
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:
return
self.regular_lr = self.get_regular_lr(runner) self.regular_lr = self.get_regular_lr(runner)
self._set_lr(runner, self.regular_lr) self._set_lr(runner, self.regular_lr)
......
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