pytorch转caffe2 之 onnx转caffe2报错的解决方法 ValueError: Don't know how to translate op Unsqueeze

目标：将 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