data_loading

# create the transformer for put the data to the tensor
transforms = transforms.Compose([
    transforms.ToTensor()
])

# load the train data - if not exists download it
train_dataset_py = torchvision.datasets.FashionMNIST(root='/content/drive/My Drive/article/data/',
                                             train=True, 
                                             transform=transforms,
                                             download=True)

test_dataset_py = torchvision.datasets.FashionMNIST(root='/content/drive/My Drive/article/data/',
                                             train=False, 
                                             transform=transforms,
                                             download=True)