add abbr to dataset argument

khanrc · Sep 21, 2017 · be7aea2 · be7aea2
1 parent 681dc34
commit be7aea2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/eval.py b/eval.py
@@ -14,7 +14,7 @@ def build_parser():
     models_str = ' / '.join(config.model_zoo)
     parser.add_argument('--model', help=models_str, required=True) 
     parser.add_argument('--name', help='default: name=model')
-    parser.add_argument('--dataset', help='CelebA / LSUN', required=True)
+    parser.add_argument('--dataset', '-D', help='CelebA / LSUN', required=True)
     parser.add_argument('--sample_size', '-N', help='# of samples. It should be a square number. (default: 16)',
         default=16, type=int)
 

diff --git a/train.py b/train.py
@@ -17,7 +17,7 @@ def build_parser():
     models_str = ' / '.join(config.model_zoo)
     parser.add_argument('--model', help=models_str, required=True) # DRAGAN, CramerGAN
     parser.add_argument('--name', help='default: name=model')
-    parser.add_argument('--dataset', help='CelebA / LSUN', required=True)
+    parser.add_argument('--dataset', '-D', help='CelebA / LSUN', required=True)
     parser.add_argument('--ckpt_step', default=5000, help='# of steps for saving checkpoint (default: 5000)', type=int)
     parser.add_argument('--renew', action='store_true', help='train model from scratch - \
         clean saved checkpoints and summaries', default=False)