diff options
Diffstat (limited to 'options/dataset_options.py')
| -rw-r--r-- | options/dataset_options.py | 6 |
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( |