提交 40609576 编写于 作者: H HydrogenSulfate

fix(dali): add shuffle to prob_list in ExternalSource_RandomIdentity

上级 b22752e2
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册