最近在做深度模型训练,测试模型的时候出现了RuntimeError:The model and loaded state dict do not match exactly.的报错,当时网上查找了一下原因,没怎么找到解决方案,突然想到,可能会不会是默认设置的原因。因为有好几个模型,所以我是先将所有的模型都先训练一遍,然后再回头进行测试。在训练完最后一个模型的时候我回到第一个模型进行测试,就出现了以下报错。
然后我就尝试修改了一下__init__.py里导入的模型和模型文件名,与准备测试的模型一致。
同时修改了utils-->configs.py里的默认设置。
再次进行测试,没有出现问题。