我遇到了一个问题，我在 pytorch 0.4.1 中训练我的模型，但我找不到将其转换为 caffe 模型的工具。
如何使用pytorch 0.4.1 model 到init pytorch 0.2.0？
或者如何将pytorch 0.4.1 model 转换为caffe model？
【问题讨论】：
Caffe2还是Caffe？
标签： deep-learning pytorch pycaffe
假设您使用Caffe2，您可以使用ONNX 将在一个 AI 框架中训练的模型转换为另一个（有一些限制）。在您的情况下，您需要将您的 PyTorch 训练模型导出到 ONNX 模型，然后将 ONNX 模型导入到 Caffe2 模型。
按照这些教程进行操作：
PyTorch 到 ONNX 导出：link
ONNX 到 Caffe2 导入：link
教程非常简单，您不需要编写太多代码。但是在将PyTorch 模型导出到ONNX 时存在一些限制。
检查此链接以获取有关Pytorch to ONNX 的深入教程。
如果您有任何疑问/问题，请在下方评论。
编辑：试试这个repo，我已经在它正在工作的玩具模型上对其进行了测试。
【讨论】：
EDIT。