error load_state_dict in val.py

leeyeehoo / CSRNet-pytorch

CSRNet: Dilated Convolutional Neural Networks for Understanding the Highly Congested Scenes

642 stars 259 forks source link

error load_state_dict in val.py #46

Open SummerHuiZhang opened 5 years ago

SummerHuiZhang commented 5 years ago

Thank you for your sharing. It's really a outstanding work. There is an error when I test my model after training, happened in line "model.load_state_dict(checkpoint['checkpoint.pth.tar']) " of val.ipynb.

It said "KeyError: 'checkpoint.pth.tar' " Could you please give me some advice?