[TensorRT实战] (1) 转换pytorch模型,一点经验之谈与例子
最近一段时间一直在负责转模型为tensorRT,鉴于我们部门的模型现在基本上都是pytorch的了,其他框架暂时不考虑。这里总结一 下经验与方法,并放一个使用networkAPI转换 resnet18的例子,这个例子是我做另一个网络模型转换时的其中的骨干网络,其他部分差不太多,不过因为是工作中的内容,就不放出来了,这里使用resnet18为例。 对一个pytorch模型,通常可以使用官方的方法,把它转成onnx,并使用TensorRT官方的,trtexec工具转成模型...
原创
2020-09-25 18:11:42 ·
1344 阅读 ·
1 评论