github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-06 04:52:13 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactoring

Browse source
This commit is contained in:
Colombo 2020-03-08 23:19:04 +04:00
parent 121c0cfc0f
commit eda6433936
5 changed files with 9 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

6
samplelib/SampleGeneratorBase.py
View file

@ -6,11 +6,7 @@ You can implement your own SampleGenerator
class SampleGeneratorBase(object):
def __init__ (self, samples_path, debug=False, batch_size=1):
if samples_path is None:
raise Exception('samples_path is None')
self.samples_path = Path(samples_path)
def __init__ (self, debug=False, batch_size=1):
self.debug = debug
self.batch_size = 1 if self.debug else batch_size
self.last_generation = None