diff options
Diffstat (limited to 'options/dataset_options.py')
| -rw-r--r-- | options/dataset_options.py | 44 |
1 files changed, 44 insertions, 0 deletions
diff --git a/options/dataset_options.py b/options/dataset_options.py index 3ba6452..4b523c7 100644 --- a/options/dataset_options.py +++ b/options/dataset_options.py @@ -99,6 +99,50 @@ class DatasetOptions(BaseOptions): help='exit immediately if set to true (used to interrupt)' ) + ## LOAD A NEW SEQUENCE + + self.parser.add_argument( + '--load-sequence', + action='store_true', + help='instruction to reload the sequence after training the next image' + ) + + self.parser.add_argument( + '--sequence-name', + type=str, + default='woodscaled_4', + help='which sequence to load' + ) + + self.parser.add_argument( + '--sequence-offset', + type=int, + default=0, + help='initial offset into sequence' + ) + + ## LOAD A NEW CHECKPOINT + + self.parser.add_argument( + '--load-checkpoint', + action='store_true', + help='instruction to reload the checkpoint after training the next image' + ) + + self.parser.add_argument( + '--checkpoint', + type=str, + default='messi', + help='which checkpoint to load' + ) + + self.parser.add_argument( + '--epoch', + type=str, + default='latest', + help='which epoch to load' + ) + ## IMAGE FILTERS ### RECURSION |
