模型转换出现的问题
收藏回复举报
模型转换出现的问题
t('forum.solved') 已解决
发表于2024-09-03 16:16:13
0 查看

在用mindspore加载模型保存文件的时候,发现出现了

这样缺少SiLU,今天尝试过2.2.10,2.2.14,2.3.0rc1,2.3.1都不行。

代码主要部分为

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.")

上面是模型加载的一些信息,应该是在模型构建的配置文件里convnormact隐性使用了这个激活函数。这是mindspore不支持这些激活函数吗,还是说我的代码有些问题(不同版本略有微调,但总体思路如上)

本帖最后由 匿名用户2024/10/08 14:47:08 编辑

我要发帖子