CSRNet-pytorch icon indicating copy to clipboard operation
CSRNet-pytorch copied to clipboard

error load_state_dict in val.py

Open SummerHuiZhang opened this issue 5 years ago • 0 comments

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?

SummerHuiZhang avatar May 18 '19 14:05 SummerHuiZhang