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.