Computer Vision Classification (Image Classification with simple model or CNN Model Implementation) from Scratch with PyTorch Taking data from FashionMNIST and then classify