fix(dali): add shuffle to prob_list in ExternalSource_RandomIdentity

2022-12-06 00:39:06 +08:00 · 2022-12-06 00:39:06 +08:00 · 4060957669
parent b22752e2ea
commit 4060957669
1 changed files with 1 additions and 0 deletions
--- a/ppcls/data/dataloader/dali.py
+++ b/ppcls/data/dataloader/dali.py
@ -529,6 +529,7 @@ class ExternalSource_RandomIdentity(object):
        if self.shuffle:
            seed = self.shard_id * 12345 + self.epoch
            np.random.RandomState(seed).shuffle(self.label_list)
+            np.random.RandomState(seed).shuffle(self.prob_list)
            self.epoch += 1
        return self