mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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

pull/459/head
Jintao Lin 2020-08-03 11:55:51 +08:00 committed by GitHub
parent 9de04d2284
commit 71b59a8da0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mmcv/runner/hooks/lr_updater.py
View File

@ -110,13 +110,14 @@ class LrUpdaterHook(Hook):
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:
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
self.regular_lr = self.get_regular_lr(runner)
self._set_lr(runner, self.regular_lr)