三亩地 三亩地SAN MU DI · CODE DIARY
ARTICLE DETAIL

日记详情

真实记录编程学习的某一天,欢迎挑你感兴趣的翻一翻。

【RT-DETR实战】117、华为昇腾CANN部署探索:从模型转换到推理踩坑实录

【RT-DETR实战】117、华为昇腾CANN部署探索:从模型转换到推理踩坑实录

一、模型转换的第一道坎

上周三晚上十点,盯着屏幕上第N次转换失败的日志,我意识到RT-DETR要上昇腾平台,第一个拦路虎根本不是算法本身,而是模型转换这道工序。

PyTorch训练好的.pth文件在昇腾310B芯片上跑起来,中间隔着一整套工具链——CANN的ATC工具。

# 典型的转换命令,这里踩过坑atc--model=rtdetr.onnx \--framework=5\--output=rtdetr_ascend \-
← 返回列表