@chetan_patil
Oh! Makes sense. My understanding was incorrect. I thought torch.load would load the entire model along with the parameters, architecture and the weights.
It appears, it will only load the weight and not anything else. Is that correct?
Edit: Just to add, I don’t plan to resume training. I intend to save the model and use it for a separate training with same parameters. Think of it like using a saved model with weights etc for a larger run and more samples.