ONNX转paddle新的报错 - Githubissues

PaddlePaddle / X2Paddle

Deep learning model converter for PaddlePaddle. (『飞桨』深度学习模型转换工具)

http://www.paddlepaddle.org/

Apache License 2.0

735 stars 164 forks source link

ONNX转paddle新的报错 #845

Open yangshurong opened 2 years ago

yangshurong commented 2 years ago

上次我发的报错是因为参数是cpu模式下的，这次的我在cuda模式下，将pytorch模型转换成了onnx然后再尝试使用x2paddle时产生了新的报错

问题描述

错误信息
错误截图

TENSOR_TYPE_TO_NP_TYPE[item.type.tensor_type.elem_type],

转换模型后用处
- [x] 转换预训练参数，再使用 Paddle 进行模型开发
模型来源链接：https://pan.baidu.com/s/1YGOvAN89S0nSwSNozGRATg?pwd=ijxt 提取码：ijxt
应用场景 micronet pytorch版的预训练模型，将用paddledetection来复现论文做目标检测
版本信息 PaddlePaddle => 2.2.2: X2Paddle => 1.3.8: 来源框架版本(PyTorch/TF/ONNX/Caffe) =>onnx1.12.0:
您的联系方式(邮箱/微信/电话) 15066146083@163.com

wjj19950828 commented 2 years ago

@yangshurong 麻烦用netron看一下onnx具体输入的shape，然后输入准确的shape而不是1再试试

另外，X2Paddle还是以推理模型转换为主，预训练模型建议训练好之后再进行转换

若要进行论文复现，需要预训练模型，建议将组网与Torch对齐后，直接将参数映射过来～