I basically disected the code from the MNIST CNN tutorial from Ax and put my network into it, the Optimisation process was able to run when I explicitly define the hyperparamete