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

SAE: added pixel loss option.

Browse source
This commit is contained in:
iperov 2019-01-25 09:56:40 +04:00
parent 72646becd1
commit ef3cf392c5
2 changed files with 23 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

8
models/ModelBase.py
View file

@ -106,14 +106,8 @@ class ModelBase(object):
self.options.pop('target_epoch')
self.batch_size = self.options['batch_size']
self.sort_by_yaw = self.options['sort_by_yaw']
if not self.sort_by_yaw:
self.options.pop('sort_by_yaw')
self.sort_by_yaw = self.options['sort_by_yaw']
self.random_flip = self.options['random_flip']
if self.random_flip:
self.options.pop('random_flip')
self.src_scale_mod = self.options['src_scale_mod']
if self.src_scale_mod == 0: