(mindspore_py39_2.2.10) PS C: \Users\24305\PycharmProj ects\yolov5_6\t rain> python onnx.py --config • /yolov5s.yaml
Parse Model, args: nearest, keep str type
Parse Model, args: nearest, keep str type
Traceback (most recent call last):
    File "C: \Users\24305\PycharmProjects\yolov5_6\train\onnx.py", line 86, in <module>
        ms.export(model, input_tensor, file_name='lenet' , file_format='ONNX')
    File "C: \Users\24305\.anaconda\envs\mindspore_py39_2.2.10\lib\site-packages\mindsporeltrain\serialization.py", line 1659, in export
        _export(net, file_name, file_format, *inputs, **kwargs)
    File "c: \Users \24305\.anacondalenvs\mindspore_py39_2.2.10\lib\site-packages\mindsporeltrain\serialization.py", line 1713, in _export
        _save_onnx(net, file_name, *inputs, **kwargs)
    File "C: \Users\24305\.anaconda\envs\mindspore_py39_2.2.10\lib|site-packages\mindsporeltrain\serialization.py", line 1770, in _save_onnx
        onnx_stream = _executor ._get_func_graph_proto (net, graph_id)
    File "c: \Users\24305\.anaconda\envs\mindspore_py39_2.2.10\liblsite-packages\mindsporelcommonlapi.py", line 1667, in _get-func_graph_proto
        return self._graph_executon .get_func_graph_proto(exec_id, ir_type, incremental)
RuntimeError: Can not find key SiLU in convert map. Exporting SiLU operator is not yet supported.

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