• 中文
  • 3.8.1

    • 模型导出工具

    算法包提供两种模型导出工具，支持ONNX和HBIR模型导出。

    • tools/export_onnx.py：ONNX模型导出工具，导出的ONNX模型只可用于可视化，不支持推理。
    • tools/export_hbir.py：HBIR模型导出工具。

    导出ONNX模型

    使用方法

    执行脚本 tools/export_onnx.py 并指定配置文件路径，脚本将自动读取配置中的模型、输入、导出路径等信息，完成 ONNX 模型导出。

    参数介绍

    参数描述
    --config, -cconfig文件路径。

    使用示例

    以 resnet50_imagenet 模型为例：

    python3 tools/export_onnx.py --config configs/classification/resnet50_imagenet.py

    导出HBIR模型

    使用方法

    执行脚本 tools/export_hbir.py 并指定配置文件路径，脚本将自动读取配置中的模型、输入、导出路径等信息，完成 HBIR 模型导出。

    注意

    如果要将参考算法包提供的模型部署在nash-p ，nash-b或nash-h上面，需要传入对应的march参数。

    参数介绍

    参数描述
    --config, -cconfig文件路径。
    --save-pathhbir保存的路径。
    --marchBPU march，可选值为nash-b-lite、nash-b、nash-b-plus、nash-e、nash-m、nash-h、nash-p。
    --with-check是否使用示例输入检查 mlir 模型的前向推理过程，可选值为True和False，建议您使用默认值False。
    --debug是否将日志级别设置为 DEBUG 级，可选值为True和False，建议您使用默认值False。

    使用示例

    以 resnet50_imagenet 模型为例：

    python export_hbir.py --config configs/classification/resnet50_imagenet.py --save-path ./output/hbir --march nash-p

    config配置文件的详细介绍可参考config配置章节。