No importer registered for op: nan_to_num. Attempting to import as plugin.
When I convert on orin, I get an error, my torch version is 1.12 [03/31/2023-10:54:43] [TRT] [V] Searching for input: onnx::Greater_2181 [03/31/2023-10:54:43] [TRT] [V] node_of_onnx::And_2182 [Greater] inputs: [onnx::Greater_2180 -> (4, 1, 6, 2500, 1)[FLOAT]], [onnx::Greater_2181 -> ()[FLOAT]], [03/31/2023-10:54:43] [TRT] [V] Registering layer: onnx::Greater_2181 for ONNX node: onnx::Greater_2181 [03/31/2023-10:54:43] [TRT] [V] Registering layer: node_of_onnx::And_2182 for ONNX node: node_of_onnx::And_2182 [03/31/2023-10:54:43] [TRT] [V] Registering tensor: onnx::And_2182 for ONNX tensor: onnx::And_2182 [03/31/2023-10:54:43] [TRT] [V] node_of_onnx::And_2182 [Greater] outputs: [onnx::And_2182 -> (4, 1, 6, 2500, 1)[BOOL]], [03/31/2023-10:54:43] [TRT] [V] Parsing node: node_of_aten::nan_to_num_2183 [And] [03/31/2023-10:54:43] [TRT] [V] Searching for input: onnx::And_2175 [03/31/2023-10:54:43] [TRT] [V] Searching for input: onnx::And_2182 [03/31/2023-10:54:43] [TRT] [V] node_of_aten::nan_to_num_2183 [And] inputs: [onnx::And_2175 -> (4, 1, 6, 2500, 1)[BOOL]], [onnx::And_2182 -> (4, 1, 6, 2500, 1)[BOOL]], [03/31/2023-10:54:43] [TRT] [V] Registering layer: node_of_aten::nan_to_num_2183 for ONNX node: node_of_aten::nan_to_num_2183 [03/31/2023-10:54:43] [TRT] [V] Registering tensor: aten::nan_to_num_2183 for ONNX tensor: aten::nan_to_num_2183 [03/31/2023-10:54:43] [TRT] [V] node_of_aten::nan_to_num_2183 [And] outputs: [aten::nan_to_num_2183 -> (4, 1, 6, 2500, 1)[BOOL]], [03/31/2023-10:54:43] [TRT] [V] Parsing node: node_of_onnx::Cast_2187 [nan_to_num] [03/31/2023-10:54:43] [TRT] [V] Searching for input: aten::nan_to_num_2183 [03/31/2023-10:54:43] [TRT] [V] node_of_onnx::Cast_2187 [nan_to_num] inputs: [aten::nan_to_num_2183 -> (4, 1, 6, 2500, 1)[BOOL]], [optional input, not set], [optional input, not set], [optional input, not set], [03/31/2023-10:54:43] [TRT] [I] No importer registered for op: nan_to_num. Attempting to import as plugin. [03/31/2023-10:54:43] [TRT] [I] Searching for plugin: nan_to_num, plugin_version: 1, plugin_namespace: ERROR: Failed to parse the ONNX file. In node 1482 (importFallbackPluginImporter): UNSUPPORTED_NODE: Assertion failed: creator && "Plugin not found, are the plugin name, version, and namespace correct?"
See #27
@xll426 have you solved this issue yet?