search

ItayElam / SegmentAnything-TensorRT

31 stars 1 forks source link

op LayerNormalization error #2

Closed babar414 closed 10 months ago

babar414 commented 10 months ago

Hi, Thank you for your effort and sharing the code. While trying to run it on my computer with 3090, I run into this error:

[11/21/2023-08:18:18] [I] [TRT] No importer registered for op: LayerNormalization. Attempting to import as plugin. [11/21/2023-08:18:18] [I] [TRT] Searching for plugin: LayerNormalization, plugin_version: 1, plugin_namespace: [11/21/2023-08:18:18] [E] [TRT] parsers/onnx/ModelImporter.cpp:726: While parsing node number 3 [LayerNormalization -> "/blocks.0/norm1/LayerNormalization_output_0"]: [11/21/2023-08:18:18] [E] [TRT] parsers/onnx/ModelImporter.cpp:727: --- Begin node --- [11/21/2023-08:18:18] [E] [TRT] parsers/onnx/ModelImporter.cpp:728: input: "/Add_output_0" input: "blocks.0.norm1.weight" input: "blocks.0.norm1.bias" output: "/blocks.0/norm1/LayerNormalization_output_0" name: "/blocks.0/norm1/LayerNormalization" op_type: "LayerNormalization" attribute { name: "axis" i: -1 type: INT } attribute { name: "epsilon" f: 1e-06 type: FLOAT }

babar414 commented 10 months ago

Fixed the error by upgrading the tensorrt from version 8.5 to 8.6.

ItayElam commented 10 months ago

I'm glad to hear the issue is resolved. just if anyone else encounters this issue: opset 17 is required for LayerNormalization used in the network and therefor a TensorRT version that supports opset 17 - 8.6 and above is required.