YOLOv4-MultiTask
YOLOv4-MultiTask copied to clipboard
loading weight error
你好,我在运行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",............
你好,我在运行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 试试