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

little fix

Browse source
This commit is contained in:
iperov 2019-02-16 21:14:15 +04:00
parent 642b88a8d3
commit 3a9d450281
4 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
models/Model_DF/Model.py
View file

@ -15,7 +15,8 @@ class Model(ModelBase):
#override
def onInitializeOptions(self, is_first_run, ask_override):
if is_first_run or ask_override:
self.options['pixel_loss'] = self.options['pixel_loss'] = input_bool ("Use pixel loss? (y/n, ?:help skip: n/default ) : ", False, help_message="Default DSSIM loss good for initial understanding structure of faces. Use pixel loss after 20k epochs to enhance fine details and remove face jitter.")
def_pixel_loss = self.options.get('pixel_loss', False)
self.options['pixel_loss'] = input_bool ("Use pixel loss? (y/n, ?:help skip: n/default ) : ", def_pixel_loss, help_message="Default DSSIM loss good for initial understanding structure of faces. Use pixel loss after 20k epochs to enhance fine details and remove face jitter.")
else:
self.options['pixel_loss'] = self.options.get('pixel_loss', False)