From 0df8723fd140c893ec4177ffe2e53b9ce3db3b4e Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Wed, 5 Sep 2018 11:58:03 +0200 Subject: augment script --- options/dataset_options.py | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'options/dataset_options.py') 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( -- cgit v1.2.3-70-g09d2