From f82bc7031eb4d3d03f653a9f329629ebfaab4c5d Mon Sep 17 00:00:00 2001 From: iperov Date: Thu, 10 Jan 2019 19:29:17 +0400 Subject: [PATCH] fix SAE --- models/Model_SAE/Model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/Model_SAE/Model.py b/models/Model_SAE/Model.py index e46c27b..541c78b 100644 --- a/models/Model_SAE/Model.py +++ b/models/Model_SAE/Model.py @@ -39,13 +39,13 @@ class SAEModel(ModelBase): default_style_power = 100 if is_first_run else self.options['face_style_power'] 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', default_style_power) + 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'] 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', default_style_power) + self.options['bg_style_power'] = self.options.get('bg_style_power', 100) default_ae_dims = 256 if self.options['archi'] == 'liae' else 512