Add saving after some number of epochs