From 43064e641457fd996cc44fbc40f3e30126e7be4f Mon Sep 17 00:00:00 2001 From: Gilles Rochefort Date: Thu, 15 Aug 2019 19:48:56 -0400 Subject: [PATCH] Add a gamma parameter in CycleGAN train_options.py --- models/networks.py | 2 +- options/train_options.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/models/networks.py b/models/networks.py index ae088f6e78e..c3a30d20bee 100644 --- a/models/networks.py +++ b/models/networks.py @@ -54,7 +54,7 @@ def lambda_rule(epoch): return lr_l scheduler = lr_scheduler.LambdaLR(optimizer, lr_lambda=lambda_rule) elif opt.lr_policy == 'step': - scheduler = lr_scheduler.StepLR(optimizer, step_size=opt.lr_decay_iters, gamma=0.1) + scheduler = lr_scheduler.StepLR(optimizer, step_size=opt.lr_decay_iters, gamma=opt.lr_gamma) elif opt.lr_policy == 'plateau': scheduler = lr_scheduler.ReduceLROnPlateau(optimizer, mode='min', factor=0.2, threshold=0.01, patience=5) elif opt.lr_policy == 'cosine': diff --git a/options/train_options.py b/options/train_options.py index 8b8ebfba6cb..380aa298517 100644 --- a/options/train_options.py +++ b/options/train_options.py @@ -35,6 +35,7 @@ def initialize(self, parser): parser.add_argument('--pool_size', type=int, default=50, help='the size of image buffer that stores previously generated images') parser.add_argument('--lr_policy', type=str, default='linear', help='learning rate policy. [linear | step | plateau | cosine]') parser.add_argument('--lr_decay_iters', type=int, default=50, help='multiply by a gamma every lr_decay_iters iterations') + parser.add_argument('--lr_gamma', type=float, default=0.1, help='gamma value used with lr_decay_iters') self.isTrain = True return parser