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

fix

Browse source
This commit is contained in:
seranus 2021-12-06 23:30:36 +01:00
commit cdd3b08b85
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
samplelib/SampleGeneratorFace.py
View file

@ -142,4 +142,6 @@ class SampleGeneratorFace(SampleGeneratorBase):
for i in range(len(x)): for i in range(len(x)):
batches[i].append ( x[i] ) batches[i].append ( x[i] )
filenames.append(sample.filename)
yield ([ np.array(batch) for batch in batches], filenames) yield ([ np.array(batch) for batch in batches], filenames)