diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-04-22 20:25:09 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-04-22 20:25:09 -0700 |
| commit | 9137b1146f9b0edb14d7c4421ed7abed1583b025 (patch) | |
| tree | f643838e27813345408b26dd55ddd36454a9f249 /options | |
| parent | 6c347282993d2e2db91b376d3113efa3774c3a22 (diff) | |
add comment for python 2/3
Diffstat (limited to 'options')
| -rw-r--r-- | options/train_options.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/options/train_options.py b/options/train_options.py index b241863..981126f 100644 --- a/options/train_options.py +++ b/options/train_options.py @@ -10,8 +10,8 @@ class TrainOptions(BaseOptions): self.parser.add_argument('--continue_train', action='store_true', help='continue training: load the latest model') self.parser.add_argument('--phase', type=str, default='train', help='train, val, test, etc') self.parser.add_argument('--which_epoch', type=str, default='latest', help='which epoch to load? set to latest to use latest cached model') - self.parser.add_argument('--niter', type=int, default=100, help='# of iter at starting learning rate') - self.parser.add_argument('--niter_decay', type=int, default=100, help='# of iter to linearly decay learning rate to zero') + 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') |