1 系统环境
硬件环境(Ascend/GPU/CPU): Ascend/GPU/CPU
MindSpore版本: mindspore=2.2.10
执行模式(PyNative/ Graph):不限
Python版本: Python=3.8
操作系统平台: linux
2 报错信息
2.1 问题描述
在用mindspore的如下脚本加载模型保存文件的时候,出现了如下报错。
2.2 脚本信息
2.3 报错信息
3 根因分析
******此处由用户填写******
4 解决方案
******此处由用户填写******
包含文字方案和最终脚本代码
请将正确的脚本打包并上传附件
1 系统环境
硬件环境(Ascend/GPU/CPU): Ascend/GPU/CPU
MindSpore版本: mindspore=2.2.10
执行模式(PyNative/ Graph):不限
Python版本: Python=3.8
操作系统平台: linux
2 报错信息
2.1 问题描述
在用mindspore的如下脚本加载模型保存文件的时候,出现了如下报错。
2.2 脚本信息
ckpt_file = 'C:/Users/24305/PycharmProjects/yolov5_6/train/runs/2024.09.03-10.11.55/weights/yolov5s-6_8.ckpt' param_dict = load_checkpoint(ckpt_file) load_param_into_net(model, param_dict) input_tensor = Tensor(np.ones([1, 3, 256, 256]).astype(np.float32)) ms.export(model, input_tensor, file_name='lenet', file_format='ONNX') print("ONNX model exported successfully.")2.3 报错信息
3 根因分析
******此处由用户填写******
4 解决方案
******此处由用户填写******
包含文字方案和最终脚本代码
请将正确的脚本打包并上传附件