Samples elements sequentially, always in the same order. Arguments: data_source (Dataset): dataset to sample from
| 20 | |
| 21 | |
| 22 | class SequentialSampler(Sampler): |
| 23 | """Samples elements sequentially, always in the same order. |
| 24 | |
| 25 | Arguments: |
| 26 | data_source (Dataset): dataset to sample from |
| 27 | """ |
| 28 | |
| 29 | def __init__(self, data_source): |
| 30 | self.data_source = data_source |
| 31 | |
| 32 | def __iter__(self): |
| 33 | return iter(range(len(self.data_source))) |
| 34 | |
| 35 | def __len__(self): |
| 36 | return len(self.data_source) |
| 37 | |
| 38 | |
| 39 | class RandomSampler(Sampler): |