Solved by AnBucquet in post #2 
              
              
                Hey, 
You should look at this tutorial. I think it’s what you’re looking for. 
But basically you need to create your own dataset class like: 
class Dataset(Dataset):
    """dataset."""

    def __init__(self, csv_file, root_dir, transform=None):
        """
        Args:
            csv_file (string…