From 079da5c02fd99ef35d7cad0e20c2924b7c2bcffd Mon Sep 17 00:00:00 2001 From: junyanz Date: Sun, 18 Feb 2018 22:53:26 -0500 Subject: fix test_model & add timer for data loader & rename identity loss --- options/train_options.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'options') diff --git a/options/train_options.py b/options/train_options.py index f4627ce..3d05a2b 100644 --- a/options/train_options.py +++ b/options/train_options.py @@ -21,12 +21,12 @@ 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('--lambda_identity', type=float, default=0.5, + help='use identity mapping. Setting lambda_identity other than 0 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 lambda_identity = 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.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.parser.add_argument('--identity', type=float, default=0.5, - 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.isTrain = True -- cgit v1.2.3-70-g09d2