Missing key(s) in state

tech2022-07-07  234

这个错误是在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)

最新回复(0)