yuanyao366 / PRP

Apache License 2.0
40 stars 10 forks source link

Error When Loading Pretrained Models #8

Open shahd-seddik opened 3 years ago

shahd-seddik commented 3 years ago

I am facing an error when loading the state_dict. It states that there are missing keys ("linear.weight", "linear.bias") as well as unexpected keys. See attached image for error details. state_dict

Please note that this particular screenshot is from when I try C3D. Similar error happens when I try to load R3D or R(2+1)D. Please advise. Thank you in advance.

yuanyao366 commented 3 years ago

After you run train_predict.py and get the pretrained model, you should run the ft_classfy.py first, and then use test_classify to test. Please refer to README.md