debug

nnet/xsets.py

@@ -106,6 +106,10 @@ class SideSampler(torch.utils.data.Sampler):
 
 
     def __iter__(self):
+        g = torch.Generator()
+        g.manual_seed(self.seed + self.epoch)
+        numpy.random.seed(self.seed + self.epoch)
+
         # Generate batches per speaker
         straight = numpy.arange(self.spk_count)
         indices = numpy.ones((self.samples_per_speaker, self.spk_count), dtype=numpy.int) * straight
@@ -136,9 +140,6 @@ class SideSampler(torch.utils.data.Sampler):
         # we want to convert the speaker indexes into segment indexes
         self.index_iterator = numpy.zeros_like(batch_matrix)
 
-        g = torch.Generator()
-        g.manual_seed(self.seed + self.epoch)
-
         # keep track of next segment index to sample for each speaker
         for idx, value in enumerate(batch_matrix):
             if self.segment_cursors[value] > len(self.labels_to_indices[value]) - 1: