Hy K.Frank. Thank you for your kind advice. I am importing Fashion MNIST data via this code. Can it be a problem:
train_data=datasets.ImageFolder(“C:/Users/mashood/Desktop/tootoo/Datasets/Fashion_MNIST/train”,transform=train_transform)
test_data = datasets.ImageFolder(“C:/Users/mashood/Desktop/tootoo/Datasets/Fashion_MNIST/test”,transform=test_transform)