操作指南

参数说明

表1 参数说明

参数

参数说明

取值示例

-i

--input
  • 要进行转换的原始工程目录。
  • 必选。

/home/username/x2mindspore

-o

--ouput
  • 脚本转换结果文件输出路径。会在该路径下输出带有x2ms后缀的目录。
  • 必选。

/home/username/x2mindspore_output

-f

--framework
  • 原始脚本的框架，目前仅支持pytorch。
  • 必选。

pytorch

-h

--help
  • 显示帮助信息。
  • 可选。

-

执行转换

  1. 进入脚本转换工具所在路径。

    cd Ascend-cann-toolkit安装目录/ascend-toolkit/{version}/{arch}-linux/toolkit/tools/x2mindspore/

  2. 执行脚本转换工具。

    ./run_x2mindspore.sh -i 原始脚本路径 -o 脚本转换结果输出路径 -f 原始脚本的框架

  3. 完成脚本转换。

    脚本转化后，在上一步执行脚本转换工具指定的脚本转换结果文件输出路径中会生成转换后的xxx_x2ms脚本文件、x2ms_adapter适配层文件、supported_api.csv支持算子列表文件和unsupported_api.csv不支持算子列表文件。

  4. 在执行转换后的模型文件前请先将输出的工程路径加入环境变量PYTHONPATH中。
父主题： X2MindSpore