mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-06 13:02:15 -07:00
removing trailing spaces
This commit is contained in:
parent
fa4e579b95
commit
a3df04999c
61 changed files with 2110 additions and 2103 deletions
|
@ -4,22 +4,21 @@ from pathlib import Path
|
|||
You can implement your own SampleGenerator
|
||||
'''
|
||||
class SampleGeneratorBase(object):
|
||||
|
||||
|
||||
|
||||
|
||||
def __init__ (self, samples_path, debug, batch_size):
|
||||
if samples_path is None:
|
||||
raise Exception('samples_path is None')
|
||||
|
||||
|
||||
self.samples_path = Path(samples_path)
|
||||
self.debug = debug
|
||||
self.batch_size = 1 if self.debug else batch_size
|
||||
|
||||
self.batch_size = 1 if self.debug else batch_size
|
||||
|
||||
#overridable
|
||||
def __iter__(self):
|
||||
#implement your own iterator
|
||||
return self
|
||||
|
||||
|
||||
def __next__(self):
|
||||
#implement your own iterator
|
||||
return None
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue