Big~Newbie

I want to save the trainned model after every epoch, I save the `local model` defined in this code and also extra data .i.e step....every time reload from the previous...