From 7800d516596f1a25986b458cddf8b8785bcc7df8 Mon Sep 17 00:00:00 2001 From: junyanz Date: Fri, 6 Oct 2017 10:46:43 -0700 Subject: support nc=1, add new leaerning rate policy and new initialization --- options/train_options.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'options/train_options.py') diff --git a/options/train_options.py b/options/train_options.py index a595017..f8a0ff6 100644 --- a/options/train_options.py +++ b/options/train_options.py @@ -21,4 +21,7 @@ class TrainOptions(BaseOptions): self.parser.add_argument('--lambda_B', type=float, default=10.0, help='weight for cycle loss (B -> A -> B)') self.parser.add_argument('--pool_size', type=int, default=50, help='the size of image buffer that stores previously generated images') self.parser.add_argument('--no_html', action='store_true', help='do not save intermediate training results to [opt.checkpoints_dir]/[opt.name]/web/') + self.parser.add_argument('--lr_policy', type=str, default='lambda', help='learning rate policy: lambda|step|plateau') + self.parser.add_argument('--lr_decay_iters', type=int, default=50, help='multiply by a gamma every lr_decay_iters iterations') + self.isTrain = True -- cgit v1.2.3-70-g09d2