saving the model out of the train method...please