search

Jungjee / RawNet

Official repository for RawNet, RawNet2, and RawNet3
MIT License
357 stars 55 forks source link

About the pretrained model #13

Closed taylorlu closed 3 years ago

taylorlu commented 3 years ago

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 commented 3 years ago

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

taylorlu commented 3 years ago

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