I am new to neural networks and I am trying to develop a nn that predicts the handwritten digit given by an image. This is my code used for training:
import m