YOLOv4-MultiTask icon indicating copy to clipboard operation
YOLOv4-MultiTask copied to clipboard

loading weight error

Open LiShengG opened this issue 4 years ago • 1 comments

你好,我在运行detect_demo.py时,模型加载错误。请问这个是什么原因呢? Missing key(s) in state_dict: "total_ops", "total_params", "module_list.total_ops", "module_list.total_params", "module_list.0.total_ops", "module_list.0.total_params", "module_list.0.Conv2d.total_ops",............

LiShengG avatar Jan 05 '21 13:01 LiShengG

你好,我在运行detect_demo.py时,模型加载错误。请问这个是什么原因呢? Missing key(s) in state_dict: "total_ops", "total_params", "module_list.total_ops", "module_list.total_params", "module_list.0.total_ops", "module_list.0.total_params", "module_list.0.Conv2d.total_ops",............

我这边运行没有问题也,你在 load_state_dict()的参数中加上 strict=False 试试

Huangdebo avatar Jan 06 '21 08:01 Huangdebo