From ee5d38e287e3dda25fa159baa7232d6f6588267a Mon Sep 17 00:00:00 2001 From: junyanz Date: Sat, 9 Sep 2017 02:42:21 -0700 Subject: update options --- options/train_options.py | 1 + 1 file changed, 1 insertion(+) (limited to 'options/train_options.py') diff --git a/options/train_options.py b/options/train_options.py index a595017..1064115 100644 --- a/options/train_options.py +++ b/options/train_options.py @@ -19,6 +19,7 @@ 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