| 4 | |
| 5 | |
| 6 | class DataLoader(torch.utils.data.DataLoader): |
| 7 | def __init__( |
| 8 | self, |
| 9 | dataset, |
| 10 | batch_size=1, |
| 11 | shuffle=False, |
| 12 | sampler=None, |
| 13 | batch_sampler=None, |
| 14 | num_workers=0, |
| 15 | collate_fn=time_aware_collate, |
| 16 | pin_memory=False, |
| 17 | drop_last=False, |
| 18 | timeout=0, |
| 19 | worker_init_fn=None, |
| 20 | ): |
| 21 | super().__init__( |
| 22 | dataset, |
| 23 | sampler=sampler, |
| 24 | shuffle=shuffle, |
| 25 | batch_size=batch_size, |
| 26 | drop_last=drop_last, |
| 27 | pin_memory=pin_memory, |
| 28 | timeout=timeout, |
| 29 | num_workers=num_workers, |
| 30 | worker_init_fn=worker_init_fn, |
| 31 | batch_sampler=batch_sampler, |
| 32 | collate_fn=collate_fn, |
| 33 | ) |
no outgoing calls
no test coverage detected