目标:将 pytorch模型 转为 onnx模型 再转为 caffe2模型,得到两个.pb文件
pytorch转onnx可通过torch.onnx._export函数实现。
onnx转caffe2有两种方法:
第一种方法:使用命令行
参考: https://github.com/onnx/tutorials/blob/master/tutorials/OnnxCaffe2Import.ipynb
截图如下:
参考上述方法可以实现onnx模型到caffe2模型的转换。
第二种方法:写文件
参考:https://my.oschina.net/u/1431433/blog/2878668
截图如下:
以及:https://pyto