Missing key(s) in state_dict: “total_ops“, “total_params“, “module_list.total_ops“,

这个错误是在yolov3项目上遇见的,因为使用了htop,去获取整个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,不用htop,卸载htop

2,在model.load_state_dict(state_dice,strick=False)

你可能感兴趣的:(pytorch)