运行pytorch2onnx.py报错

polarisZhao / PFLD-pytorch

PFLD pytorch Implementation

798 stars 197 forks source link

运行pytorch2onnx.py报错 #33

Closed DWCTOD closed 4 years ago

DWCTOD commented 4 years ago

您好，运行pytorch2onnx.py报错了，

错误输出一大段内容，最后是： ====> check onnx model... 段错误 (核心已转储)

不知是什么原因，求大佬help

github-luffy commented 4 years ago

我之前遇到过但还是直接转ncnn，是可以用的；你可以试试转ncnn 然后看有没有用

github-luffy commented 4 years ago

如果想训练68点的话可以直接点击PFLD_68points_Pytorch

DWCTOD commented 4 years ago

如果想训练68点的话可以直接点击PFLD_68points_Pytorch

好的，谢谢大佬，我试一哈我尝试了一下pytorch2onnx.py，发现原来是pytorch 、onnx、onnxruntime和onnx-simplifier版本的问题经过不断尝试发现，下面的组合是可以的 pytorch=1.12.0 onnx=1.5.0 onnxruntime=0.4.0 onnx-simplifier=0.1.8