Add ability to restart training an existing model.