Skip to content
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

About the pretrained model #13

Closed
taylorlu opened this issue Nov 9, 2020 · 2 comments
Closed

About the pretrained model #13

taylorlu opened this issue Nov 9, 2020 · 2 comments

Comments

@taylorlu
Copy link

taylorlu commented Nov 9, 2020

Hi, when I test the model by using pretrained model rawnet2_best_weights.pt, it throws an error
in Missing key(s) in state_dict: "block0.0.frm.fc.weight", "block0.0.frm.fc.bias", "block1.0.frm.fc.weight", "block1.0.frm.fc.bias", "block2.0.frm.fc.weight", "block2.0.frm.fc.bias", "block3.0.frm.fc.weight", "block3.0.frm.fc.bias", "block4.0.frm.fc.weight", "block4.0.frm.fc.bias", "block5.0.frm.fc.weight", "block5.0.frm.fc.bias
It seems really missing these parameters after I checked the model by Netron.
Many thanks for your reply in advance.

@Jungjee
Copy link
Owner

Jungjee commented Nov 9, 2020

Hello, have you tried loading on this file?
model_RawNet2_original_code.py

@taylorlu
Copy link
Author

taylorlu commented Nov 9, 2020

OK, it's my mistake, the original code is alll right.

@Jungjee Jungjee closed this as completed Nov 9, 2020
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants