Remove unneeded train() from NeuralNetwork.