在用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不支持这些激活函数吗,还是说我的代码有些问题(不同版本略有微调,但总体思路如上)
在用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不支持这些激活函数吗,还是说我的代码有些问题(不同版本略有微调,但总体思路如上)