PaddleClas
PaddleClas copied to clipboard
paddle-onnx-gemm操作和torch-onnx-gemm操作不一致问题
问题描述:
目前使用paddle框架训练repvgg分类模型,然后进行转onnx,之后在进行转caffe,但是发现转caffe的时候,遇到了op不支持的问题。 同样使用mmcls-torch框架训练resnet18分类模型,然后进行转哦你那些,之后进行转caffe,发现转caffe是可以的。
比对了,两个分类模型onnx拓扑图,发现均存在gemm操作。具体如下图:
考虑到caffe框架比较老,目前想通过对paddle-repvgg的gemm操作做修改来实现caffe模型的转换。
请问是否有好的建议或方法?谢谢!