Looks like it was a version problem - I trained the model in 0.4.1 and tried to load the dict in 0.4.0. Once I upgraded the pytorch version, everything worked fine.
Looks like it was a version problem - I trained the model in 0.4.1 and tried to load the dict in 0.4.0. Once I upgraded the pytorch version, everything worked fine.