search

QiShanZhang / SLSforASVspoof-2021-DF

Code for paper "Audio Deepfake Detection with Self-supervised XLS-R and SLS classifier
6 stars 0 forks source link

您好，您的代码在加载权重时模型的键依然无法对齐。 #2

Closed littleJu1y closed 1 month ago

littleJu1y commented 1 month ago

由于本人愚钝，代码依然运行失败，您能否看一下报错信息。 这是我的加载代码：target_model = Model("cuda:0") print(target_model) target_model.load_state_dict(torch.load("/home/hh/data/voice_srd/target/SLSforADD/MMpaper_model.pth",map_location='cuda:0')) 报错见附件。 nnn.txt

不胜感激！

QiShanZhang commented 1 month ago

这可能是因为我的模型权重中有预训练模型，加载方式推荐使用我的代码，预训练模型的加载使用的是fairseq的库。