PRP icon indicating copy to clipboard operation
PRP copied to clipboard

Published 20 hours ago • verified publisher icon yuanyao366

Error When Loading Pretrained Models

Open shahd-seddik opened this issue 3 years ago • 1 comments

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.

shahd-seddik avatar May 23 '21 14:05 shahd-seddik

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

yuanyao366 avatar Jun 02 '21 08:06 yuanyao366