Skip to content

Commit 7e5fffc

Browse files
committed
adding coode source
1 parent 87ed084 commit 7e5fffc

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

references/classification/sampler.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,9 @@ class RASampler(torch.utils.data.Sampler):
1010
It ensures that different each augmented version of a sample will be visible to a
1111
different process (GPU).
1212
Heavily based on 'torch.utils.data.DistributedSampler'.
13+
14+
This is borrowed from the DeiT Repo:
15+
https://github.com/facebookresearch/deit/blob/main/samplers.py
1316
"""
1417

1518
def __init__(self, dataset, num_replicas=None, rank=None, shuffle=True):

0 commit comments

Comments
 (0)