From 2a344ccd6f80ce0435d2c58ccfd94c76dd423b1a Mon Sep 17 00:00:00 2001 From: junyanz Date: Thu, 19 Oct 2017 19:35:11 -0700 Subject: update learning rate policy --- models/networks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'models/networks.py') diff --git a/models/networks.py b/models/networks.py index 0b8938d..19169c5 100644 --- a/models/networks.py +++ b/models/networks.py @@ -88,7 +88,7 @@ def get_norm_layer(norm_type='instance'): def get_scheduler(optimizer, opt): if opt.lr_policy == 'lambda': 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) + lr_l = 1.0 - max(0, epoch - opt.niter + 1) / float(opt.niter_decay + 1) return lr_l scheduler = lr_scheduler.LambdaLR(optimizer, lr_lambda=lambda_rule) elif opt.lr_policy == 'step': -- cgit v1.2.3-70-g09d2