Animating neural network training