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.ip...
原创
2019-04-05 17:55:05 ·
2035 阅读 ·
1 评论