基于香橙派AIpro将开源框架模型转换为昇腾模型

原文约3300字,阅读约需8分钟。发表于:

昇腾张量编译器是昇腾模型转换工具,它可以将开源框架的网络模型转换为昇腾AI处理器支持的模型文件,用于后续的模型推理。

本文介绍了如何使用华为云的昇腾AI处理器和ATC工具将开源框架模型转换为昇腾模型。ATC工具支持从Caffe、ONNX、TensorFlow和MindSpore模型转换为昇腾AI处理器支持的.om格式。文章还介绍了ATC工具的基本使用方法和高级特性,如将模型转换为json格式、自定义模型的输入输出数据类型以及设置动态BatchSize和动态分辨率。

相关推荐 去reddit讨论