Based on PyTorch Docs Results after 60 ish epochs This would work better with convolutions but when creating this i had not learnt abotu CNNs yet