sinan_mutlu:

To load the parameters directly from a file after instantiating the model is there any specific thing that I have to do?


Not that I’m aware of and the model.load_state_dict() call should raise a warning, if missing or unexpected keys were found.
Could you post a small code snippet, which shows how you are restoring the model (in the non-working case) and how you are fixing this issue?