提交 41041092 编写于 作者: W weishengyu

add comment for pk sampler

上级 ee1bc18f
......@@ -81,6 +81,7 @@ class PKSampler(DistributedBatchSampler):
def __iter__(self):
label_per_batch = self.batch_size // self.sample_per_label
if self.shuffle:
# It's not accurate literally, but it helps in some dataset.
np.random.RandomState(self.epoch).shuffle(self.label_list)
for i in range(len(self)):
batch_index = []
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册