| 19 | |
| 20 | |
| 21 | class NumpyLoader(data.DataLoader): |
| 22 | def __init__(self, dataset, batch_size=1, |
| 23 | shuffle=False, sampler=None, |
| 24 | batch_sampler=None, num_workers=0, |
| 25 | pin_memory=False, drop_last=False, |
| 26 | timeout=0, worker_init_fn=None): |
| 27 | super(self.__class__, self).__init__(dataset, |
| 28 | batch_size=batch_size, |
| 29 | shuffle=shuffle, |
| 30 | sampler=sampler, |
| 31 | batch_sampler=batch_sampler, |
| 32 | num_workers=num_workers, |
| 33 | collate_fn=numpy_collate, |
| 34 | pin_memory=pin_memory, |
| 35 | drop_last=drop_last, |
| 36 | timeout=timeout, |
| 37 | worker_init_fn=worker_init_fn) |
| 38 | |
| 39 | |
| 40 | class MotionDataset(data.Dataset): |