From e39d1a2619a9902f67c403406ec66a9f5ba7f43a Mon Sep 17 00:00:00 2001 From: iperov Date: Thu, 10 Jan 2019 23:01:33 +0400 Subject: [PATCH] fixes --- models/ModelBase.py | 4 ++-- models/Model_SAE/Model.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/models/ModelBase.py b/models/ModelBase.py index 446ff50..8e746a9 100644 --- a/models/ModelBase.py +++ b/models/ModelBase.py @@ -64,7 +64,7 @@ class ModelBase(object): force_best_gpu_idx = input_int("Which GPU idx to choose? ( skip: system choice ) : ", -1) if self.epoch == 0 or ask_override: - default_write_preview_history = False if self.epoch == 0 else self.options['batch_size'] + default_write_preview_history = False if self.epoch == 0 else self.options.get('write_preview_history',False) self.options['write_preview_history'] = input_bool("Write preview history? (y/n ?:help skip:n/default) : ", default_write_preview_history, help_message="Preview history will be writed to _history folder.") else: self.options['write_preview_history'] = self.options.get('write_preview_history', False) @@ -75,7 +75,7 @@ class ModelBase(object): self.options['target_epoch'] = self.options.get('target_epoch', 0) if self.epoch == 0 or ask_override: - default_batch_size = 0 if self.epoch == 0 else self.options['batch_size'] + default_batch_size = 0 if self.epoch == 0 else self.options.get('batch_size',0) self.options['batch_size'] = max(0, input_int("Batch_size (?:help skip:0/default) : ", default_batch_size, help_message="Larger batch size is always better for NN's generalization, but it can cause Out of Memory error. Tune this value for your videocard manually.")) else: self.options['batch_size'] = self.options.get('batch_size', 0) diff --git a/models/Model_SAE/Model.py b/models/Model_SAE/Model.py index 541c78b..85f23c7 100644 --- a/models/Model_SAE/Model.py +++ b/models/Model_SAE/Model.py @@ -36,13 +36,13 @@ class SAEModel(ModelBase): self.options['lighter_encoder'] = self.options.get('lighter_encoder', False) if is_first_run or ask_override: - default_style_power = 100 if is_first_run else self.options['face_style_power'] + default_style_power = 100 if is_first_run else self.options.get('face_style_power', 100) self.options['face_style_power'] = np.clip ( input_int("Face style power (0..100 ?:help skip:%d) : " % (default_style_power), default_style_power, help_message="How fast NN will learn dst face style during generalization of src and dst faces."), 0, 100 ) else: self.options['face_style_power'] = self.options.get('face_style_power', 100) if is_first_run or ask_override: - default_style_power = 100 if is_first_run else self.options['bg_style_power'] + default_style_power = 100 if is_first_run else self.options.get('bg_style_power', 100) self.options['bg_style_power'] = np.clip ( input_int("Background style power (0..100 ?:help skip:%d) : " % (default_style_power), default_style_power, help_message="How fast NN will learn dst background style during generalization of src and dst faces."), 0, 100 ) else: self.options['bg_style_power'] = self.options.get('bg_style_power', 100)