From fef48ca10031efbd769382e45b376f2de98e34e2 Mon Sep 17 00:00:00 2001 From: Aleksey Smolenchuk Date: Tue, 16 Apr 2024 09:13:22 -0700 Subject: [PATCH] Fix InstanceNormalization_TRT plugin initialization --- utilities.py | 1 + 1 file changed, 1 insertion(+) diff --git a/utilities.py b/utilities.py index 1ff6499..5b922c1 100644 --- a/utilities.py +++ b/utilities.py @@ -234,6 +234,7 @@ def build( update_output_names=None, ): print(f"Building TensorRT engine for {onnx_path}: {self.engine_path}") + trt.init_libnvinfer_plugins(None, "") p = [Profile()] if input_profile: p = [Profile() for i in range(len(input_profile))]