模型编译
【免费下载链接】docs该仓库用于维护cann公共文档项目地址: https://gitcode.com/cann/docs
以运行用户登录开发环境。使用ATC工具将开源框架的网络模型(如ONNX、TensorFlow等)编译成*.om模型文件。
此处以转换ONNX模型为例给出命令示例,执行命令的用户需具有命令中相关路径的可读、可写权限。
atc --model=$HOME/module/resnet50*.onnx --framework=5 --output=$HOME/module/out/onnx_resnet50 --soc_version=<soc_version>各参数的解释如下,详细取值说明及约束说明请参见《ATC离线模型编译工具》。
- --model:ResNet-50网络的模型文件的路径。
- --framework:原始框架类型。5表示ONNX。
- --output:resnet50.om模型文件的路径。请注意,记录保存该om模型文件的路径,后续开发应用时需要使用。
- --soc_version:AI处理器的版本。_<soc_version>_请根据实际情况替换。
注意事项:
- 如果模型转换时,提示有不支持的算子,请先参见《Ascend C算子开发指南》完成自定义算子开发,再重新编译模型。
- 如果模型转换时,提示有算子编译相关问题,但根据报错信息无法定位问题、需要联系技术支持时(您可以获取日志后单击Link联系技术支持。),则需设置DUMP_GE_GRAPH、DUMP_GRAPH_LEVEL环境变量,再重新转换模型,收集模型转换过程中各个阶段的图描述信息。关于环境变量以及图描述信息的说明,请参见《ATC离线模型编译工具》中的“参考 > dump图详细信息”。
【免费下载链接】docs该仓库用于维护cann公共文档项目地址: https://gitcode.com/cann/docs
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考