mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

remove trick code

pull/1273/head
weishengyu 2021-09-29 10:59:16 +08:00
parent 41041092b5
commit c4d54dd0e0
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ppcls/data/dataloader/pk_sampler.py
View File

@ -80,10 +80,7 @@ class PKSampler(DistributedBatchSampler):
def __iter__(self): def __iter__(self):
label_per_batch = self.batch_size // self.sample_per_label label_per_batch = self.batch_size // self.sample_per_label
if self.shuffle: for _ in range(len(self)):
# 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 = [] batch_index = []
batch_label_list = np.random.choice( batch_label_list = np.random.choice(
self.label_list, self.label_list,