You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
when i reload the model_best.pth.tar, i got this error:
=> loading checkpoint './model_best.pth.tar'
Traceback (most recent call last):
File "prune.py", line 37, in
model.load_state_dict(checkpoint['state_dict'])
File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/module.py", line 522, in load_state_dict
.format(name))
KeyError: 'unexpected key "module.feature.0.weight" in state_dict'
how can i solve this problem???
The text was updated successfully, but these errors were encountered:
when i reload the model_best.pth.tar, i got this error:
=> loading checkpoint './model_best.pth.tar'
Traceback (most recent call last):
File "prune.py", line 37, in
model.load_state_dict(checkpoint['state_dict'])
File "/usr/local/lib/python3.5/dist-packages/torch/nn/modules/module.py", line 522, in load_state_dict
.format(name))
KeyError: 'unexpected key "module.feature.0.weight" in state_dict'
how can i solve this problem???
The text was updated successfully, but these errors were encountered: