当前位置: 首页 > news >正文

pytorch格式转华为昇腾的om格式

具体的转换过程为:pytorch => onnx => om

pytorch转onnx比较简单,教程有一大堆了,下面给出 onnx 转 om 格式的过程。

atc --model=models/model.onnx     --framework=5     --output=models/model     --input_shape="input_ids:1,512;attention_mask:1,512;token_type_ids:1,512"     --soc_version=Ascend910B2

atc命令中各参数的含义如下:

–model:model的ONNX模型文件存储路径。
–framework:原始框架类型,5表示ONNX。
–output:离线om模型的路径以及文件名。
–input_shape:执行推理时模型输入数据的shape,这个在转onnx的时候也需要。
–soc_version:昇腾AI处理器的型号。
请在服务器执行npu-smi info命令进行查询,在查询到的“Name”前增加Ascend信息,例如“Name”对应取值为xxxyy,实际配置的<soc_version>值为Ascendxxxyy。

比较关键的点有两个:

  1. input_shape要正确输入,参考我的命令示例。ATC工具对于多输入的语法是:input1:shape1;input2:shape2;input3:shape3
  2. soc_version需要通过npu-smi info命令进行查询,比如我的机器就是Ascend910B2,
    在这里插入图片描述
http://www.dtcms.com/a/304755.html

相关文章:

  • 移动语义和右值引用有什么关系?
  • Prometheus-1--什么是Prometheus?
  • Leetcode——475. 供暖器
  • Python - property
  • 学习笔记-中华心法问答系统的性能提升
  • pnpm 入门与实践指南
  • 字节序详解
  • webpack 原理及使用
  • 纸板制造胶工艺学习1
  • FlinkCDC实现多数据源数据集的增量更新实战
  • 磁悬浮转子同频振动:自适应陷波器设计与稳定性深度解析(附MATLAB代码)
  • 8.项目起步(2)
  • 本土化DevOps实践新篇章:Gitee引领企业高效协作新时代
  • 自学嵌入式 day37 HTML
  • Go 客户端玩转 ES|QL API 直连与 Mapping Helpers 实战详解
  • 做副业,要事不过三
  • CentOS 7部署Redis
  • 如何理解js中的副作用
  • 沪铝本周想法
  • docker docker与swarm入门笔记
  • 云原生周刊:2025年的服务网格
  • 【Linux基础知识系列】第七十篇 - 了解swap分区的概念
  • 【Linux篇】补充:消息队列和systemV信号量
  • 【图像噪点消除】——图像预处理(OpenCV)
  • 以太坊十年:智能合约与去中心化的崛起
  • 第5课 sort函数
  • Couchbase 详解
  • HBase 详解
  • 使用jQuery时的注意事项
  • Java 大视界 -- Java 大数据机器学习模型在金融信用评级模型优化与信用风险动态管理中的应用(371)