diff options
| author | Taesung Park <taesung_park@berkeley.edu> | 2017-04-27 01:28:53 -0700 |
|---|---|---|
| committer | Taesung Park <taesung_park@berkeley.edu> | 2017-04-27 01:28:53 -0700 |
| commit | e5b2fd6d36b4297c4314478e88820cd10943d192 (patch) | |
| tree | aa6bf853e47a858b3e9d8a5ed051377f9b0b6415 /options/train_options.py | |
| parent | af7420fc67f1a69349f8155bb85a3536314e377b (diff) | |
1. Added one_direction_test_model that generates the outputs in only one direction
2. Changed the option naming from ntrain to max_dataset_size
Diffstat (limited to 'options/train_options.py')
| -rw-r--r-- | options/train_options.py | 1 |
1 files changed, 0 insertions, 1 deletions
diff --git a/options/train_options.py b/options/train_options.py index 981126f..4b4eac3 100644 --- a/options/train_options.py +++ b/options/train_options.py @@ -13,7 +13,6 @@ class TrainOptions(BaseOptions): self.parser.add_argument('--niter', type=int, default=200, help='# of iter at starting learning rate') self.parser.add_argument('--niter_decay', type=int, default=0, help='# of iter to linearly decay learning rate to zero') self.parser.add_argument('--beta1', type=float, default=0.5, help='momentum term of adam') - self.parser.add_argument('--ntrain', type=int, default=float("inf"), help='# of examples per epoch.') self.parser.add_argument('--lr', type=float, default=0.0002, help='initial learning rate for adam') 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)') |