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

Update lr_updater.py (#95) 

* Update lr_updater.py

"Add min_lr to PolyLrUpdaterHook, so that it can anneal to min_lr. When min_lr=0, it behaves as usual. "

* Update lr_updater.py
pull/91/head^2
HarryHan 2019-07-30 23:23:54 +08:00 committed by Kai Chen
parent eae81c1e86
commit 5a5c94b68b
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -138,8 +138,9 @@ class ExpLrUpdaterHook(LrUpdaterHook):
class PolyLrUpdaterHook(LrUpdaterHook):
def __init__(self, power=1., **kwargs):
def __init__(self, power=1., min_lr=0., **kwargs):
self.power = power
self.min_lr = min_lr
super(PolyLrUpdaterHook, self).__init__(**kwargs)
def get_lr(self, runner, base_lr):
@ -149,7 +150,8 @@ class PolyLrUpdaterHook(LrUpdaterHook):
else:
progress = runner.iter
max_progress = runner.max_iters
return base_lr * (1 - progress / max_progress)**self.power
coeff = (1 - progress / max_progress)**self.power
return (base_lr - self.min_lr) * coeff + self.min_lr
class InvLrUpdaterHook(LrUpdaterHook):