Would be nice to be able to restart training with reinitialised weights, quickly and easily. Not sure how best to do this right now.