Onnx模型转换报错,可能时版本不兼容

onnx模型转换时报错:

onnx.onnx_cpp2py_export.checker.ValidationError: Node () has input size 4 not in range [min=1, max=1].

==> Context: Bad node spec: input: “576” input: “578” input: “579” input: “577” output: “580” op_type: “Slice”

有可能是版本不兼容的问题,但因为torch=1.2.0不兼容torchvision=0.6,我没用requirements.txt的环境;我的环境是torch=1.4.0,onnx=1.4.1.

请问这个问题有办法解决吗

@yuming 你已经转成onnx,再使用转换工具的么,那就跟pytroch的环境没啥关系了

这个问题已经解决了,

  1. 在转onnx时要设置opset_version=9;
  2. 同时,我的网络中用了插值F.interpolate,要把参数size=(H,W)根据网络结果写出具体的数,这样可以避开对H,W推断的算子,本质上就是把动态图写成静态图,可以解决我现在的问题。

这样Step 0就可以过了,但是Step 1出现了另外的问题 :sweat_smile:,麻烦帮忙看一下