这个错误是在yolov3项目上遇见的,因为使用了thop,去获取整个model 的mac,所以会导致total_params以及total_ops被注册到网络中,又因为你的pt文件中根本没有这个key,所以才会导致miss错误。
使用下面两种方式查看网络的参数(被优化器可优化参数,不被优化器优化的参数)
#学习参数
for name,param in model.named_parameters():
print(name)
#常量参数或者是不能被优化器优化的参数
for name,buf in model.named_buffers():
print(name)
解决办法有两种:
1,不用thop,卸载thop
2,在model.load_state_dict(state_dict,strict=False)