hb_config_generator工具

hb_config_generator 是用于支持您获取模型编译最简yaml配置文件、包含全部参数默认值的yaml配置文件的工具。其中:

  • 最简yaml配置文件中会包含您可以用于快速验证模型转换过程能否正常执行的最简参数配置。

  • 包含全部参数默认值的yaml配置文件中会包含模型编译时具有默认值的所有参数。

使用方法

  1. 生成最简yaml配置文件:

    hb_config_generator --simple-yaml
  2. 生成基于模型信息的最简yaml配置文件:

    hb_config_generator --simple-yaml --model model.onnx --march nash-e
  3. 包含全部参数默认值的yaml配置文件:

    hb_config_generator --full-yaml
  4. 生成基于模型信息的全部参数默认值yaml配置文件:

    hb_config_generator --full-yaml --model model.onnx --march nash-e

命令行参数

参数名称                            参数说明
-h, --help显示帮助信息。
-s, --simple-yaml生成最简yaml配置文件,用于您快速验证模型转换能否正常进行。
-f, --full-yaml生成包含全部参数默认值的yaml配置文件。
-m, --modelCaffe或ONNX模型文件。
-p, --proto用于指定Caffe模型prototxt文件。
--marchBPU的微架构。使用S100处理器需设置为 nash-e ,使用S100P处理器需设置为 nash-m

输出内容说明

hb_config_generator 命令会根据您的配置生成对应的最简yaml配置文件或包含全部参数默认值的yaml配置文件。

注意

请注意,在使用包含完整参数默认值的yaml文件前需要将该文件内的默认值修改为您当前实际情况的对应值。