Wraps an arbitrary object with __len__ and __getitem__ into a pytorch dataset
| 313 | |
| 314 | |
| 315 | class WrappedDataset(Dataset): |
| 316 | """Wraps an arbitrary object with __len__ and __getitem__ into a pytorch dataset""" |
| 317 | |
| 318 | def __init__(self, dataset): |
| 319 | self.data = dataset |
| 320 | |
| 321 | def __len__(self): |
| 322 | return len(self.data) |
| 323 | |
| 324 | def __getitem__(self, idx): |
| 325 | return self.data[idx] |
| 326 | |
| 327 | |
| 328 | def worker_init_fn(_): |