search

mindspore-lab / mindyolo

A toolbox of yolo models and algorithms based on MindSpore
Apache License 2.0
101 stars 42 forks source link

加载yolov8预训练模型，做训练，权重网络不匹配 #344

Open Salary-only-17k opened 2 months ago

Salary-only-17k commented 2 months ago

RuntimeError: For 'load_param_into_net', model.model.22.cv3.0.2.weight in the argument 'net' should have the same shape as model.model.22.cv3.0.2.weight in the argument 'parameter_dict'. But got its shape (10, 128, 1, 1) in the argument 'net' and shape (80, 128, 1, 1) in the argument 'parameter_dict'.

Salary-only-17k commented 2 months ago

我测试了yolov8s yolov7tiny yolov5n都不能加载？？？？

WongGawa commented 2 months ago

感谢反馈。请提供一下详细信息。 执行命令是什么，并且权重从哪里获取得到的呢？