diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-08-26 18:28:29 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-08-26 18:28:29 -0700 |
| commit | 8bc6778456c86a8b9d88362efb195635d2a0dac3 (patch) | |
| tree | 0a068f150820d3d31ef1609634ba340cefe719ef /options/train_options.py | |
| parent | f085c5c977af48cfaed3c61b0a8b9c951ad88337 (diff) | |
add epoch_count
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 345f619..a595017 100644 --- a/options/train_options.py +++ b/options/train_options.py @@ -9,6 +9,7 @@ class TrainOptions(BaseOptions): self.parser.add_argument('--save_latest_freq', type=int, default=5000, help='frequency of saving the latest results') self.parser.add_argument('--save_epoch_freq', type=int, default=5, help='frequency of saving checkpoints at the end of epochs') self.parser.add_argument('--continue_train', action='store_true', help='continue training: load the latest model') + self.parser.add_argument('--epoch_count', type=int, default=1, help='the starting epoch count, we save the model by <epoch_count>, <epoch_count>+<save_latest_freq>, ...') 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') |