From 7cdae10bcfa1c9dba7a41eaaff082c4ed647e984 Mon Sep 17 00:00:00 2001 From: zengshao0622 Date: Thu, 22 Dec 2022 03:38:38 +0000 Subject: [PATCH] fix pksampler prob list shuffle bug --- ppcls/data/dataloader/pk_sampler.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/ppcls/data/dataloader/pk_sampler.py b/ppcls/data/dataloader/pk_sampler.py index cbbc0919f..11d1ac8e6 100644 --- a/ppcls/data/dataloader/pk_sampler.py +++ b/ppcls/data/dataloader/pk_sampler.py @@ -104,6 +104,8 @@ class PKSampler(DistributedBatchSampler): rank = dist.get_rank() np.random.RandomState(rank * self.total_epochs + self.epoch).shuffle(self.label_list) + np.random.RandomState(rank * self.total_epochs + + self.epoch).shuffle(self.prob_list) self.epoch += 1 label_per_batch = self.batch_size // self.sample_per_id