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