I am trying to write a training loop in MXNet with the Gluon API and write a validation loop to evaluate a trained network.
I am currently trying to define a network. I