这个错误是因为model.load
中的model
的问题。
你下载到了别人的模型参数文件,但是这个model你必须先按照别人的代码中那个model构造,而不能是你的model。
也就是说,单单下载别人的模型参数文件时没有用的,还1.得知道那个人的
class Model(nn.Module):
是怎么定义的。
2.以及初始化模型的时候传入的那些超参数是什么。知道了这两个之后,就可以:
model=Model(那个人的超参)
model.load(那个人的模型参数文件)
然后就成功了,我们免去了训练过程,得到了训练好的model,然后可以开始自己测试。
也就是说,那个人可能很多代码,但是上述2个代码部分是务必要有的,其他可以扔了。