Hi, if you pass an input model, at the end of the training you will have your output model. Why do you want to fetch the input model from the previous step?