模型编译
以运行用户登录开发环境。使用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>请根据实际情况替换。
注意事项:
父主题: 模型推理