Saving during training - how does it work?