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.
请问如何让解决。
[注:日志文件见附件,报错在文件后面部分。
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.
请问如何让解决。
[注:日志文件见附件,报错在文件后面部分。