Datasets
Models
Diff of
/datasets/dataloader_prefetch.py
[000000]
..
[03464c]
Switch to unified view
| a | b/datasets/dataloader_prefetch.py | ||
|---|---|---|---|
| 1 | from torch.utils.data import DataLoader |
||
| 2 | from prefetch_generator import BackgroundGenerator |
||
| 3 | |||
| 4 | |||
| 5 | class DataLoaderPrefetch(DataLoader): |
||
| 6 | def __iter__(self): |
||
| 7 | return BackgroundGenerator(super().__iter__()) |