diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-10-19 19:19:39 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-10-19 19:19:39 -0700 |
| commit | f87530a15cdf92064297825b8b7a9e23da87f6a5 (patch) | |
| tree | d63357c16f3bfaec710e6c7960b5564a3823b79c | |
| parent | f34719c53419d283586c47ad1905b8144a89547a (diff) | |
fix small issue in learning rate policy
| -rw-r--r-- | models/networks.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/models/networks.py b/models/networks.py index 2df58fe..0b8938d 100644 --- a/models/networks.py +++ b/models/networks.py @@ -87,8 +87,8 @@ def get_norm_layer(norm_type='instance'): def get_scheduler(optimizer, opt): if opt.lr_policy == 'lambda': - def lambda_rule(epoch): - lr_l = 1.0 - max(0, epoch - opt.niter) / float(opt.niter_decay+1) + def lambda_rule(epoch): # epoch ranges from [1, opt.niter+opt.niter_decay] + lr_l = 1.0 - max(0, epoch - opt.niter) / float(opt.niter_decay) return lr_l scheduler = lr_scheduler.LambdaLR(optimizer, lr_lambda=lambda_rule) elif opt.lr_policy == 'step': |