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

added safeguards

Browse source
This commit is contained in:
Brigham Lysenko 2019-08-14 16:36:19 -06:00
commit 3a2bc5ed0e
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
models/ModelBase.py
View file

@ -566,8 +566,10 @@ class ModelBase(object):
if self.batch_size == 1: if self.batch_size == 1:
self.paddle = 'ping' self.paddle = 'ping'
if self.paddle == 'ping': if self.paddle == 'ping':
self.save()
self.set_batch_size(self.batch_size + 1) self.set_batch_size(self.batch_size + 1)
else: else:
self.save()
self.set_batch_size(self.batch_size - 1) self.set_batch_size(self.batch_size - 1)
self.iter += 1 self.iter += 1