CNN_FashionMNIST-with-pytorch