【RT-DETR实战】117、华为昇腾CANN部署探索:从模型转换到推理踩坑实录
一、模型转换的第一道坎
上周三晚上十点,盯着屏幕上第N次转换失败的日志,我意识到RT-DETR要上昇腾平台,第一个拦路虎根本不是算法本身,而是模型转换这道工序。
PyTorch训练好的.pth文件在昇腾310B芯片上跑起来,中间隔着一整套工具链——CANN的ATC工具。
# 典型的转换命令,这里踩过坑atc--model=rtdetr.onnx \--framework=5\--output=rtdetr_ascend \-