PaddleClas icon indicating copy to clipboard operation
PaddleClas copied to clipboard

paddle-onnx-gemm操作和torch-onnx-gemm操作不一致问题

Open tensorflowt opened this issue 2 years ago • 0 comments

问题描述:

目前使用paddle框架训练repvgg分类模型,然后进行转onnx,之后在进行转caffe,但是发现转caffe的时候,遇到了op不支持的问题。 同样使用mmcls-torch框架训练resnet18分类模型,然后进行转哦你那些,之后进行转caffe,发现转caffe是可以的。

比对了,两个分类模型onnx拓扑图,发现均存在gemm操作。具体如下图:

repvgg

resnet18

考虑到caffe框架比较老,目前想通过对paddle-repvgg的gemm操作做修改来实现caffe模型的转换。

请问是否有好的建议或方法?谢谢!

tensorflowt avatar Jun 07 '23 03:06 tensorflowt