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

added ability to not use ping pong

Browse source
This commit is contained in:
Brigham Lysenko 2019-08-13 15:43:00 -06:00
commit 9e498c5d71
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/ModelBase.py
View file

@ -526,7 +526,7 @@ class ModelBase(object):
if self.iter == 1 and self.options.get('ping_pong', True):
self.set_batch_size(1)
self.paddle = 'ping'
elif not self.options.get('ping_pong', True):
elif not self.options.get('ping_pong', True) and self.batch_cap != self.batch_size:
self.set_batch_size(self.batch_cap)
sample = self.generate_next_sample()
iter_time = time.time()