search

Torbjorn1997 / IIRP-Net

11 stars 0 forks source link

关于mind.pt加载报错 #6

Open YingYuxinLeah opened 4 weeks ago

YingYuxinLeah commented 4 weeks ago

作者您好，打扰了！请问如果方便的话可以重新上传一下可用的模型参数吗？我这边加载您的模型参数时也是会报模型结构与加载的模型的状态字典不匹配的错误。非常感谢

Torbjorn1997 commented 1 week ago

还有其他同学提出了这个问题，我做了这样的一个修改，在我的本地可以绕过这个问题，但我没有得到反馈该方法是否可行，也许你可以试试 具体来说，我修改了原本train.py中的代码： model = vxm.iirp.IIRPNet.load(args.model, device) 将该代码改成了： model = vxm.iirp.IIRPNet(img_size) model.load_state_dict(torch.load(args.model)['model_state'], strict=False)

Torbjorn1997 commented 1 week ago

如果可行麻烦您反馈一下，我修改github上的代码，因为我本地没遇到这个问题，我也不确定这样是否能解决。