Solved by albanD in post #2 
              
              
                From the error that you linked, it seems that you’re missing a from torch.utils import data to have data.DataLoader to work properly.