change dataset naming convention and add ui model

1 files changed, 1 insertions, 0 deletions

diff --git a/options/base_options.py b/options/base_options.py
index 863c061..de831fe 100755
--- a/options/base_options.py
+++ b/options/base_options.py
@@ -15,6 +15,7 @@ class BaseOptions():
         self.parser.add_argument('--name', type=str, default='label2city', help='name of the experiment. It decides where to store samples and models')        
         self.parser.add_argument('--gpu_ids', type=str, default='0', help='gpu ids: e.g. 0  0,1,2, 0,2. use -1 for CPU')                       
         self.parser.add_argument('--checkpoints_dir', type=str, default='./checkpoints', help='models are saved here')
+        self.parser.add_argument('--model', type=str, default='pix2pixHD', help='which model to use')
         self.parser.add_argument('--norm', type=str, default='instance', help='instance normalization or batch normalization')        
         self.parser.add_argument('--use_dropout', action='store_true', help='use dropout for the generator')