summaryrefslogtreecommitdiff
path: root/options/dataset_options.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-09-05 11:58:03 +0200
committerJules Laplace <julescarbon@gmail.com>2018-09-05 11:58:03 +0200
commit0df8723fd140c893ec4177ffe2e53b9ce3db3b4e (patch)
treefff66ee63aaff689503969f6223dc90891ad224f /options/dataset_options.py
parentfd3198c0c799e7943f7f27758e97670535c94979 (diff)
augment script
Diffstat (limited to 'options/dataset_options.py')
-rw-r--r--options/dataset_options.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/options/dataset_options.py b/options/dataset_options.py
index 29e07c6..93fbb6d 100644
--- a/options/dataset_options.py
+++ b/options/dataset_options.py
@@ -309,6 +309,12 @@ class DatasetOptions(BaseOptions):
help='number of recursive images to generate'
)
+ self.parser.add_argument(
+ '--augment-name',
+ type=str,
+ help='name to tag epoch'
+ )
+
### GRAYSCALE
self.parser.add_argument(