网站介绍:文章浏览阅读3.3k次。在深度学习模型部署时,从pytorch转换onnx的过程中,踩了一些坑。本文总结了这些踩坑记录,希望可以帮助其他人。首先,简单说明一下pytorch转onnx的意义。在pytorch训练出一个深度学习模型后,需要在TensorRT或者openvino部署,这时需要先把Pytorch模型转换到onnx模型之后再做其它转换。因此,在使用pytorch训练深度学习模型完成后,在TensorRT或者openvino或者opencv和onnxruntime部署时,pytorch模型转onnx这一步是必不可少的。接_pytorch转onnx python变量导致中断
- 链接地址:https://blog.csdn.net/Mrrunsen/article/details/120753021
- 链接标题:模型部署翻车记:pytorch转onnx踩坑实录_pytorch转onnx python变量导致中断-CSDN博客
- 所属网站:blog.csdn.net
- 被收藏次数:8519
- 网站标签:pytorch转onnx python变量导致中断