From 5b04cc047e4174f617b676869e545621903274ca Mon Sep 17 00:00:00 2001 From: leno3003 Date: Tue, 29 Mar 2022 13:07:07 +0200 Subject: [PATCH] check if pkl exist to apply modifications --- mainscripts/Trainer.py | 10 ++++++---- models/ModelBase.py | 9 +++++++-- 2 files changed, 13 insertions(+), 6 deletions(-) diff --git a/mainscripts/Trainer.py b/mainscripts/Trainer.py index 1b66348..ef0ac29 100644 --- a/mainscripts/Trainer.py +++ b/mainscripts/Trainer.py @@ -84,8 +84,9 @@ def trainerThread (s2c, c2s, e, if model.get_target_iter() != 0: if is_reached_goal: io.log_info('Model already trained to target iteration. You can use preview.') - model_save() - os._exit(1) + if os.path.exists("/home/deepfake/interact_dict.pkl"): + model_save() + os._exit(1) else: io.log_info('Starting. Target iteration: %d. Press "Enter" to stop training and save model.' % ( model.get_target_iter() ) ) else: @@ -165,8 +166,9 @@ def trainerThread (s2c, c2s, e, model_save() is_reached_goal = True io.log_info ('You can use preview now.') - model_save() - os._exit(1) + if os.path.exists("/home/deepfake/interact_dict.pkl"): + model_save() + os._exit(1) need_save = False while time.time() - last_save_time >= save_interval_min*60: diff --git a/models/ModelBase.py b/models/ModelBase.py index f446efa..d06bbf9 100644 --- a/models/ModelBase.py +++ b/models/ModelBase.py @@ -295,8 +295,13 @@ 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): - 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)) + if os.path.exists("/home/deepfake/interact_dict.pkl"): + 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)) def ask_random_flip(self): default_random_flip = self.load_or_def_option('random_flip', True)