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

2019-02-11 17:18:01 +04:00
parent f8e63970d2
commit 470fb9287a
2 changed files with 24 additions and 21 deletions
@@ -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: