TrojanXu / onnxparser-trt-plugin-sample

A sample for onnxparser working with trt user defined plugins for TRT7.0
Apache License 2.0
166 stars 36 forks source link

Error when converting onnx to tensorrt #11

Open rsj007 opened 3 years ago

rsj007 commented 3 years ago

Hi! I have already build the grid_sampler in the Tensorrt OSS project and convert my pytorch model to onnx. However it meets an error when I convert onnx to tensorrt. The error is shown below. Could you please tell me how to solve it?

[TensorRT] ERROR: INVALID_ARGUMENT: getPluginCreator could not find plugin ScatterND version 1 In node -1 (importFallbackPluginImporter): UNSUPPORTED_NODE: Assertion failed: creator && "Plugin not found, are the plugin name, version, and namespace correct?" [TensorRT] ERROR: Network must have at least one output [TensorRT] ERROR: Network validation failed. Traceback (most recent call last): File "zed_onnx2trt.py", line 9, in trt_model = onnx2trt(model) File "/home/nvidia/.local/lib/python3.6/site-packages/volksdep/converters/onnx2trt.py", line 119, in onnx2trt trt_model = TRTModel(engine) File "/home/nvidia/.local/lib/python3.6/site-packages/volksdep/converters/base.py", line 50, in init self.context = self.engine.create_execution_context() AttributeError: 'NoneType' object has no attribute 'create_execution_context'