diff --git a/models/ModelBase.py b/models/ModelBase.py index d669ac0..81be71f 100644 --- a/models/ModelBase.py +++ b/models/ModelBase.py @@ -295,23 +295,21 @@ class ModelBase(object): self.choose_preview_history = io.input_bool("Randomly choose new image for preview history", False, help_message="Preview image history will stay stuck with old faces if you reuse the same model on different celebs. Choose no unless you are changing src/dst to a new person") def ask_target_iter(self, default_value=0): - print("Path in ask_target_iter") - cmd = "pwd" - os.system(cmd) - #answer_filename = 'workspace/interact/interact_dict.pkl' - #if os.path.exists(answer_filename): - # with open(answer_filename, 'rb') as file: - # parameters = pickle.load(file) - # if parameters['NoInteractiveMode']: - # default_target_iter = self.load_or_def_option('target_iter', default_value + 10000) - # self.options['target_iter'] = max(0, default_target_iter + 10000) - # print("TARGET ITERATION: " + str(self.options['target_iter'])) - # else: - # default_target_iter = self.load_or_def_option('target_iter', default_value) - # self.options['target_iter'] = max(0, io.input_int("Target iteration", default_target_iter)) - #else: - default_target_iter = self.load_or_def_option('target_iter', default_value) - self.options['target_iter'] = max(0, io.input_int("Target iteration", default_target_iter)) + answer_filename = 'workspace/interact/interact_dict.pkl' + if os.path.exists(answer_filename): + with open(answer_filename, 'rb') as file: + parameters = pickle.load(file) + if parameters['NoInteractiveMode']: + print("Entro in ask_target_iter") + default_target_iter = self.load_or_def_option('target_iter', default_value + 10000) + self.options['target_iter'] = max(0, default_target_iter + 10000) + print("TARGET ITERATION: " + str(self.options['target_iter'])) + else: + default_target_iter = self.load_or_def_option('target_iter', default_value) + self.options['target_iter'] = max(0, io.input_int("Target iteration", default_target_iter)) + else: + default_target_iter = self.load_or_def_option('target_iter', default_value) + self.options['target_iter'] = max(0, io.input_int("Target iteration", default_target_iter)) def ask_random_flip(self): default_random_flip = self.load_or_def_option('random_flip', True)