Samples elements randomly, without replacement. Arguments: data_source (Dataset): dataset to sample from
| 37 | |
| 38 | |
| 39 | class RandomSampler(Sampler): |
| 40 | """Samples elements randomly, without replacement. |
| 41 | |
| 42 | Arguments: |
| 43 | data_source (Dataset): dataset to sample from |
| 44 | """ |
| 45 | |
| 46 | def __init__(self, data_source): |
| 47 | self.data_source = data_source |
| 48 | |
| 49 | def __iter__(self): |
| 50 | return iter(torch.randperm(len(self.data_source)).long()) |
| 51 | |
| 52 | def __len__(self): |
| 53 | return len(self.data_source) |
| 54 | |
| 55 | |
| 56 | class SubsetRandomSampler(Sampler): |