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

SAE: remove rare sample booster. DSSIM->MSE transition now between 5-20k epochs.

Browse source
This commit is contained in:
iperov 2019-02-11 17:18:01 +04:00
parent f8e63970d2
commit 470fb9287a
2 changed files with 24 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

1
samples/SampleGeneratorFace.py
View file

@ -54,6 +54,7 @@ class SampleGeneratorFace(SampleGeneratorBase):
generator = self.generators[self.generator_counter % len(self.generators) ]
return next(generator)
#forces to repeat these sample idxs as fast as possible
def repeat_sample_idxs(self, idxs): # [ idx, ... ]
#send idxs list to all sub generators.
for gen_sq in self.generators_sq: