--- 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__())