https://bagelmobestcheapprice.blogspot.com/2024/05/mnist-data-classification-not-working.html