github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-22 22:34:25 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

bug fix

Browse source
This commit is contained in:
Brigham Lysenko 2019-08-13 19:54:05 -06:00
commit 5ef6840593
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
models/ModelBase.py
View file

@ -198,7 +198,8 @@ class ModelBase(object):
self.sort_by_yaw = self.options.get('sort_by_yaw', False) self.sort_by_yaw = self.options.get('sort_by_yaw', False)
self.random_flip = self.options.get('random_flip', True) self.random_flip = self.options.get('random_flip', True)
if self.batch_cap == 0: if self.batch_cap == 0:
self.batch_cap = self.options['batch_cap'] = self.batch_size self.options['batch_cap'] = self.batch_size
self.batch_cap = self.options.get('batch_cap',16)
self.src_scale_mod = self.options.get('src_scale_mod', 0) self.src_scale_mod = self.options.get('src_scale_mod', 0)
if self.src_scale_mod == 0 and 'src_scale_mod' in self.options: if self.src_scale_mod == 0 and 'src_scale_mod' in self.options: