Datasets
Models
Diff of
/datasets/dataloader_prefetch.py
[000000]
..
[03464c]
Switch to side-by-side view
--- a +++ b/datasets/dataloader_prefetch.py @@ -0,0 +1,7 @@ +from torch.utils.data import DataLoader +from prefetch_generator import BackgroundGenerator + + +class DataLoaderPrefetch(DataLoader): + def __iter__(self): + return BackgroundGenerator(super().__iter__())