Thanks! I have changed it to reflect this now. But, got a different error message which says " dataloader = torch.utils.DataLoader(dataset)
AttributeError: module ‘torch.utils’ has no attribute ‘DataLoader’
I have included those lines of code, but it didn’t help:
import torch
from torchvision import datasets
import torch.utils.data