search

inisis / brocolli

Everything in Torch Fx
MIT License
341 stars 63 forks source link

KeyError: 'onnx:ScatterND' YOLOv5 conversion #39

Closed kv55 closed 2 years ago

kv55 commented 2 years ago

While trying to convert the YOLOv5 model, I'm getting KeyError for 'onnx:ScatterND', 'onnx:Range', 'onnx:Expand', 'onnx:Where' and 'onnx:Equal'. Could you please help me on how to get rid this error? I have got the same error when tried with the docker image also.

inisis commented 2 years ago

While trying to convert the YOLOv5 model, I'm getting KeyError for 'onnx:ScatterND', 'onnx:Range', 'onnx:Expand', 'onnx:Where' and 'onnx:Equal'. Could you please help me on how to get rid this error? I have got the same error when tried with the docker image also.

Hi，you can use the latest docker image

kv55 commented 2 years ago

It worked when I tried with the latest docker image. Thank you so much.