cmdz

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