Solved by localh in post #7 
              
              
                Last update: I think I have found a quick solution. Here is what I did in case it helps anyone else in the future. 
First, prepare a dataset class for CSV: 
# Create Dataset
class CSVDataset(Dataset):
    """Lazy Loading."""

    def __init__(self, csv_file, text_col, target, transform=None):
      …