diff options
Diffstat (limited to 'options/base_options.py')
| -rwxr-xr-x | options/base_options.py | 1 |
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') |