bugfix: make the epoch in MultiScaleSampler self-incrementing

PaddlePaddle · Mar 6, 2023 · b98d88e · b98d88e
1 parent acf0a17
commit b98d88e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ppcls/data/dataloader/multi_scale_sampler.py b/ppcls/data/dataloader/multi_scale_sampler.py
@@ -103,6 +103,7 @@ def __iter__(self):
                 random.seed(self.seed)
             else:
                 random.seed(self.epoch)
+            self.epoch += 1
             random.shuffle(self.img_indices)
             random.shuffle(self.img_batch_pairs)
             indices_rank_i = self.img_indices[self.rank:len(self.img_indices):