From edf910b1c1d02020b31782ab4c3b6ebf9af8c323 Mon Sep 17 00:00:00 2001 From: tingchunw Date: Tue, 30 Jan 2018 01:30:18 +0000 Subject: change dataset naming convention and add ui model --- options/base_options.py | 1 + 1 file changed, 1 insertion(+) (limited to 'options/base_options.py') 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') -- cgit v1.2.3-70-g09d2