github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-22 14:24:40 -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:49:32 -06:00
commit 49de9735d6
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
models/ModelBase.py
View file

@ -197,6 +197,8 @@ class ModelBase(object):
self.batch_cap = self.options.get('batch_cap',16)
self.sort_by_yaw = self.options.get('sort_by_yaw', False)
self.random_flip = self.options.get('random_flip', True)
if self.batch_cap == 0:
self.batch_cap = self.options['batch_cap'] = self.batch_size
self.src_scale_mod = self.options.get('src_scale_mod', 0)
if self.src_scale_mod == 0 and 'src_scale_mod' in self.options:
@ -553,6 +555,9 @@ class ModelBase(object):
if self.iter % 1000 == 0 and self.iter != 0 and self.options.get('ping_pong', True):
if self.batch_size == self.batch_cap:
self.paddle = 'pong'
if self.batch_size > self.batch_cap:
self.set_batch_size(self.batch_cap)
self.paddle = 'pong'
if self.batch_size == 1:
self.paddle = 'ping'
if self.paddle == 'ping':