Retraining a neural network