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

Question

我遇到了在import org.apache.poi.hssf.usermodel.HSSFCell; import org.apache.poi.hssf.usermodel.HSSFCellStyle; import org.apache.poi.hssf.usermodel.HSSFDataFormat; import org.apache.poi.hssf.usermodel.HSSFFont; import org.apache.poi.hssf.usermodel.HSSFRichTextString; import org.apache.poi.hssf.usermodel.HSSFRow; import org.apache.poi.hssf.usermodel.HSSFSheet; import org.apache.poi.hssf.usermodel.HSSFWorkbook; 中训练模型的问题，但是找不到用于将其转换为pytorch 0.4.1模型的工具。

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

Answer 1

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

遵循这些教程：

PyTorch到ONNX导出：link

ONNX到Caffe2导入：link

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

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

编辑：尝试使用repo，我已经在可以正常工作的玩具模型上对其进行了测试。