CachedDataSet

ovotools/pytorch/__init__.py

-from .threading_dataloader import BatchThreadingDataLoader, ThreadingDataLoader
\ No newline at end of file
+from .threading_dataloader import BatchThreadingDataLoader, ThreadingDataLoader
+from .cached_dataset import CachedDataSet
\ No newline at end of file

ovotools/pytorch/cached_dataset.py

+class CachedDataSet:
+    '''
+    provides caching dataset items in memory
+    '''
+    def __init__(self, dataset):
+        self.dataset = dataset
+        self.cache = [None] * len(dataset)
+        
+    def __len__(self):
+        return len(self.dataset)
+
+    def __getitem__(self, index):
+        if self.cache[index] is None:
+            self.cache[index] = self.dataset[index]
+        return self.cache[index]
\ No newline at end of file