-
Notifications
You must be signed in to change notification settings - Fork 255
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
KeyError: 'state_dict' #45
Comments
@jfzhang95 I run inference.py with pretrained model: |
@sdxass I run inference.py with pretrained model: |
The weight file shared by @jfzhang95 is trained on Sports-1M dataset,(It have 487 classes)its not on UCF-101,if you want to test inference code. give the sports 1m dataset labels file instead of ucf-101 labels file. |
I run inference.py with pretrained model:
Traceback (most recent call last):
File "inference.py", line 78, in
main()
File "inference.py", line 32, in main
model.load_state_dict(checkpoint['state_dict'])
KeyError: 'state_dict'
The text was updated successfully, but these errors were encountered: