Closed littleJu1y closed 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
不胜感激!
这可能是因为我的模型权重中有预训练模型,加载方式推荐使用我的代码,预训练模型的加载使用的是fairseq的库。
由于本人愚钝,代码依然运行失败,您能否看一下报错信息。 这是我的加载代码: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
不胜感激!