C++和PytorchOnnxRuntime使用方法(附代码)
Pytorch和C++OnnxRuntime使用方法写在前面Pytoch onnxruntime使用方法C++ onnxruntime使用方法写在后面写在前面最近有个需求要将Pytorch训练好的模型迁移到C++上去使用,再网上查了一些资料结合上自己的实际情况最终使用onnxruntime迁移成功。借此机会给大家分享一下,若有写的不对的地方请大家批评指正!下面给大家看看运行结果这是Pytorch的onnxruntime运行时间这是C++的onnxruntime运行时间,感觉两者速度是差不多的~可能
原创
2021-10-10 15:11:25 ·
5003 阅读 ·
20 评论