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

Question

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

如何使用init pytorch 0.2.0的pytorch 0.4.1 model

或者如何将pytorch 0.4.1 model转换为caffe model

Answer 1

假设您正在使用Caffe2，您可以使用ONNX将在一个AI框架中训练的模型转换为另一个AI框架(有一些限制)。在您的案例中，您需要将经过PyTorch训练的模型导出到ONNX模型，然后将ONNX模型导入到Caffe2模型。

请按照以下教程操作：

PyTorch到ONNX的导出：link

导入到Caffe2的ONNX：link

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

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

编辑:试试这个repo，我已经在一个玩具模型上测试过了，它正在工作。