关于RuntimeError: Error(s) in loading state_dict for VanillaKNNMT:

NJUNLP / knn-box

an easy-to-use knn-mt toolkit

MIT License

101 stars 12 forks source link

Closed txz201603604 closed 10 months ago

txz201603604 commented 10 months ago

您好，我在运行vanilla-knn-mt/inference.sh脚本后，在加载模型时出现了错误： 53BEE0C2-6E30-4acd-8C00-A5DE6FC55474 这是我的运行脚本：请问这个问题应该如何解决？

sxysxy commented 10 months ago

看起来你修改了原本infernece.sh内的参数，比如BASE_MODEL和--arch。你需要检查你的checkpoint文件存储的模型，是否能和--arch指定的模型的结构对应上