在运行SwinUnet和VisionTransformer时遇到RuntimeError，尝试检查train.py和test.py的配置一致性，以及在加载模型参数时使用`load_state_dict`函数的忽略不匹配参数。虽然能暂时解决问题，但结果不理想，寻求解决方案。
RuntimeError: Error(s) in loading state_dict for SwinUnet运行swin-unet时的错误
首先查看train.py里的num_classes、batch_size、epoch或者其他config(如lr等)是否与test.py相同，并查看生成的文件名，如果不同修改为相同。可能会成功，并且不影响结果。
不行则：
找到错误代码语句为msg = net.load_state_dict(torch.load(model_para_path))
修改成：msg = net.load_state_dict(torch.load(model_para_path),False)
RuntimeError: Error(s) in loading state_dict for VisionTransformer运行Transunet时的错误
找到错误代码语句为net.load_state_dict(torch.load(model_para_path))
按上面方法 修改为false不管用，之后将本句注释掉之后正常运行了。
以上两种方法虽然正常运行了，但是结果不尽人意，还不知道该如何修改，望指正
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
