Solved by eqy in post #6 
              
              
                You can start by taking a look at the default dataset classes: torch.utils.data — PyTorch 1.8.1 documentation 
and seeing if your data fits the map style of iterable style abstraction. The map style is usually a straightforward abstraction for many datasets as you only need to define an __getitem__ …