github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 21:12:07 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix SAE

Browse source
This commit is contained in:
iperov 2019-01-10 19:29:17 +04:00
parent bf5282f7ec
commit f82bc7031e
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
models/Model_SAE/Model.py
View file

@ -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