Solved by ptrblck in post #2 
              
              
                The main.py seems to show how to load the data. Probably you could adapt the code to load the data first and then wrap it in a Dataset. The Data Loading tutorial might help you to implement an own Dataset. 
Note that in a vanilla classification use case PyTorch expects a target filled with class ind…