From e986144cee13a921fd3ad68d564f820e8f7dd3b0 Mon Sep 17 00:00:00 2001 From: junyanz Date: Fri, 6 Oct 2017 08:54:45 -0700 Subject: move the flag identity to the base option --- options/base_options.py | 1 + options/train_options.py | 1 - 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/options/base_options.py b/options/base_options.py index d2ee44f..de8bc74 100644 --- a/options/base_options.py +++ b/options/base_options.py @@ -38,6 +38,7 @@ class BaseOptions(): self.parser.add_argument('--max_dataset_size', type=int, default=float("inf"), help='Maximum number of samples allowed per dataset. If the dataset directory contains more than max_dataset_size, only a subset is loaded.') self.parser.add_argument('--resize_or_crop', type=str, default='resize_and_crop', help='scaling and cropping of images at load time [resize_and_crop|crop|scale_width|scale_width_and_crop]') self.parser.add_argument('--no_flip', action='store_true', help='if specified, do not flip the images for data augmentation') + self.parser.add_argument('--identity', type=float, default=0.0, help='use identity mapping. Setting identity other than 1 has an effect of scaling the weight of the identity mapping loss. For example, if the weight of the identity loss should be 10 times smaller than the weight of the reconstruction loss, please set optidentity = 0.1') self.initialized = True diff --git a/options/train_options.py b/options/train_options.py index 1064115..a595017 100644 --- a/options/train_options.py +++ b/options/train_options.py @@ -19,7 +19,6 @@ class TrainOptions(BaseOptions): self.parser.add_argument('--no_lsgan', action='store_true', help='do *not* use least square GAN, if false, use vanilla GAN') self.parser.add_argument('--lambda_A', type=float, default=10.0, help='weight for cycle loss (A -> B -> A)') self.parser.add_argument('--lambda_B', type=float, default=10.0, help='weight for cycle loss (B -> A -> B)') - self.parser.add_argument('--identity', type=float, default=0.0, help='use identity mapping. Setting identity other than 1 has an effect of scaling the weight of the identity mapping loss. For example, if the weight of the identity loss should be 10 times smaller than the weight of the reconstruction loss, please set optidentity = 0.1') 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.isTrain = True -- cgit v1.2.3-70-g09d2