remove trick code
parent
41041092b5
commit
c4d54dd0e0
|
@ -80,10 +80,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)):
|
||||
for _ in range(len(self)):
|
||||
batch_index = []
|
||||
batch_label_list = np.random.choice(
|
||||
self.label_list,
|
||||
|
|
Loading…
Reference in New Issue