github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-22 06:23:20 -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:56:56 -06:00
commit 1707e54c30
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
models/ModelBase.py
View file

@ -152,7 +152,7 @@ class ModelBase(object):
self.options['paddle'] = self.options.get('paddle','ping') self.options['paddle'] = self.options.get('paddle','ping')
else: else:
self.options['batch_cap'] = self.options.get('batch_cap', 0) self.options['batch_cap'] = self.options.get('batch_cap', 16)
self.options['ping_pong'] = self.options.get('ping_pong', True) self.options['ping_pong'] = self.options.get('ping_pong', True)
if ask_sort_by_yaw: if ask_sort_by_yaw:
@ -193,7 +193,7 @@ class ModelBase(object):
if self.target_iter == 0 and 'target_iter' in self.options: if self.target_iter == 0 and 'target_iter' in self.options:
self.options.pop('target_iter') self.options.pop('target_iter')
self.batch_size = self.options.get('batch_size', 0) self.batch_size = self.options.get('batch_size', 8)
self.batch_cap = self.options.get('batch_cap',16) self.batch_cap = self.options.get('batch_cap',16)
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)