Good to hear, you’ve figured it out!



 soroosh_99:

there was a mismatch between the dictionary keys of the model and the network even though I set the strict=False.


I would not recommend to use strict=False in case you are running into unexpected mismatches while loading the state_dict, as this will ignore the keys.