minspore提供mindyolo套件训练yolov7时,训练过程中模型加载出现问题。
收藏回复举报
minspore提供mindyolo套件训练yolov7时,训练过程中模型加载出现问题。
t('forum.solved') 已解决
发表于2024-03-01 15:06:40
0 查看
1.环境配置:modelart(mindspore_1.10.0-cann_6.0.1-py_3.7-euler_2.8.3镜像)EulerOS 2.0 (SP8), CANN-6.0.1,mindspore1.10, mindyolo r0.1。 

2.数据集制作及训练过程文档(操作与文档中一致):https://github.com/mindspore-lab/mindyolo/blob/master/examples/finetune_SHWD/README.md 

3.训练过程中出现报错: 

RuntimeError For 'load_param_into_net', model.model.77.m.0.weight in the argument 'net' should have the same shape as model.model.77.m.0.weight in the argument 'parameter_dict'. But got its shape (21, 128, 1, 1) in the argument 'net' and shape (255, 128, 1, 1) in the argument 'parameter_dict'.May you need to check whether the checkpoint you loaded is correct or the batch size and so on in the 'net' and 'parameter_dict' are same. 

请问如何让解决。 

[注:日志文件见附件,报错在文件后面部分。

我要发帖子