【300I DUO】atc 模型转换报错
收藏回复举报
【300I DUO】atc 模型转换报错
t('forum.solved') 已解决
发表于2024-09-19 15:54:21
0 查看

系统环境:

os:20.04

driver: 24.1.rc2
cann: 8.0.rc2
mindie: 1.0.rc2

python:3.10
镜像链接:http://mirrors.cn-central-221.ovaijisuan.com/detail/139.html
代码链接:https://gitee.com/ascend/ModelZoo-PyTorch/tree/master/ACL_PyTorch/built-in/cv/SAM

atc 转换命令

atc \
  --framework 5 \
  --model models/encoder_modify.onnx \
  --output models/encoder_modify \
  --insert_op_conf sam.aippconfig \
  --enable_small_channel 1 \
  --soc_version "Ascend310P3"

我要发帖子