如何使用pytorch 0.4.1模型初始化pytorch 0.2.0？

Question

我遇到了一个问题，我在pytorch 0.4.1训练我的模型，但我找不到一个工具将其转换为caffe模型。

如何使用pytorch 0.4.1 model到init pytorch 0.2.0？ 或者如何将pytorch 0.4.1 model转换为caffe model？

Answer 1

假设您正在使用Caffe2，您可以使用ONNX将在一个AI框架中训练的模型转换为另一个（有一些限制）。在您的情况下，您需要将您的PyTorch训练模型导出到ONNX模型，然后将ONNX模型导入Caffe2模型。

请遵循以下教程：

PyTorch到ONNX出口：link

ONNX到Caffe2进口：link

这些教程非常简单，您不需要编写太多代码。但是将PyTorch模型导出到ONNX时存在一些限制。请查看此链接以获取有关Pytorch to ONNX的深入教程。

如果您有任何疑问/问题，请在下方评论。

编辑：尝试这个repo，我已经在它正在运行的玩具模型上进行了测试。