def gather_options(self): if not self.initialized: parser = argparse.ArgumentParser( formatter_class=argparse.ArgumentDefaultsHelpFormatter) parser = self.initialize(parser) opt, _ = parser.parse_known_args() model_option_setter = model.get_option_setter() parser = model_option_setter(parser, self.isTrain) opt, _ = parser.parse_known_args() self.parser = parser return parser.parse_args()
def gather_options(self): """Add additional model-specific options""" if not self.initialized: parser = self.initialize(self.parser) # get basic options opt, _ = parser.parse_known_args() # modify the options for different models model_option_set = model.get_option_setter(opt.model) parser = model_option_set(parser, self.isTrain) opt = parser.parse_args() return opt
def gather_options(self): """Add additional model-specific options""" if not self.initialized: self.initialize() # get basic options opt, _ = self.parser.parse_known_args() if opt.phase == 'train': self.isTrain = True self.add_train_options() else: self.isTrain = False self.add_test_options() # modify the options for different models model_option_set = model.get_option_setter(opt.model) parser = model_option_set(self.parser) opt = parser.parse_args() return opt
def gater_options(self): if not self.initialized: parser = argparse.ArgumentParser( formatter_class=argparse.ArgumentDefaultsHelpFormatter) parser = self.initialize(parser) parser.parse_args(['--epoch_count', '2']) opt, _ = parser.parse_known_args() model_name = opt.model_name model_options_setter = model.get_option_setter(model_name) parser = model_options_setter(parser, self.isTrain) opt, _ = parser.parse_known_args() # assert(opt.data_name == opt.dataset_name.split('_')[1]) self.parser = parser return parser.parse_args()
def gather_options(self): if not self.initialized: # check if it has been initialized parser = argparse.ArgumentParser( formatter_class=argparse.ArgumentDefaultsHelpFormatter) parser = self.initialize(parser) # get the basic options opt, _ = parser.parse_known_args() # modify model-related parser options # model_name = opt.model model_option_setter = model.get_option_setter() parser = model_option_setter(parser, self.isTrain) opt, _ = parser.parse_known_args() # parse again with new defaults # modify dataset-related parser options # dataset_name = opt.dataset_mode dataset_option_setter = dataset.get_option_setter() parser = dataset_option_setter(parser, self.isTrain) # save and return the parser self.parser = parser return parser.parse_args()