mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-08-19 13:09:56 -07:00
fix
This commit is contained in:
parent
ebb085e31e
commit
cdd3b08b85
1 changed files with 2 additions and 0 deletions
|
@ -142,4 +142,6 @@ class SampleGeneratorFace(SampleGeneratorBase):
|
|||
for i in range(len(x)):
|
||||
batches[i].append ( x[i] )
|
||||
|
||||
filenames.append(sample.filename)
|
||||
|
||||
yield ([ np.array(batch) for batch in batches], filenames)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue