diff options
| author | Taesung Park <taesung_park@berkeley.edu> | 2017-10-10 00:19:18 -0700 |
|---|---|---|
| committer | Taesung Park <taesung_park@berkeley.edu> | 2017-10-10 00:19:18 -0700 |
| commit | 3f8adb140c9609b62a3843cf4930b31e26da415c (patch) | |
| tree | c9cfdfbe7b62b5e686742470e87ec7bfa8c1d55f /options/train_options.py | |
| parent | fe8ea339e8f8a9dac5d202ebe29f0a578f876b86 (diff) | |
fix identity option conflict error, and also not specifying init_type in define_D
Diffstat (limited to 'options/train_options.py')
| -rw-r--r-- | options/train_options.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/options/train_options.py b/options/train_options.py index f8a0ff6..32120ec 100644 --- a/options/train_options.py +++ b/options/train_options.py @@ -23,5 +23,6 @@ class TrainOptions(BaseOptions): 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.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.isTrain = True |