pytorch模型如何用C++进行加载

2020年4月4日 下午12:13

PyTorchPyTorch的C++前端和模型部署
PyTorch直观认识torch.jit模块

总结：

1. 其实这节的内容不属于不同框架之间模型转化，讲的是针对于pytorch模型如何用C++进行加载
   • 是换语言（其实依然使用的是pytorch），而不是换框架
   • #include <torch/script.h>这个就是pytorch的C++前端接口
2. 稍后的业界的算法同学的工作流程可能就会变成这样：
   • 论文发布->PyTorch开源代码(或者自己实现)->训练模型->导出模型->载入模型(C++_Python_其他框架/其他硬件平台)