ScatterND Plugin not found while converting onnx into tensorrt model

1.2k views Asked by At

Able to convert pytorch model into onnx and validated onnx output with pytorch. it's working fine. But when I try to convert onnx into tensorrt engine, it fails at onnx parser and return this error.

[TensorRT] ERROR: INVALID_ARGUMENT: getPluginCreator could not find plugin ScatterND version 1

Onnx version : 1.6.0 Tensorrt version : 7.1.3.4

Can anyone please help here, how can I add ScatterND plugin in python and how can I use it ? Is there any other way to solve this problem ?

Thanks in advance.

0

There are 0 answers