Hot fix trt exporter

Visual-Behavior · Apr 5, 2022 · 4657ec1 · 4657ec1
1 parent 0d99941
commit 4657ec1
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/alonet/torch2trt/base_exporter.py b/alonet/torch2trt/base_exporter.py
@@ -14,7 +14,8 @@
     import tensorrt as trt
     import pycuda.driver as cuda
     prod_package_error = None
-except Exception as prod_package_error:
+except Exception as e:
+    prod_package_error = e
     pass
 
 
@@ -93,10 +94,6 @@ def __init__(
         """
         if prod_package_error is not None:
             raise prod_package_error
-        if prod_package_error is not None:
-            raise prod_package_error
-        if prod_package_error is not None:
-            raise prod_package_error
 
         if model is not None:
             assert hasattr(model, "tracing") and model.tracing, "Model must be instantiated with tracing=True"