diff --git a/trt_utilities.py b/trt_utilities.py
index 44deefa..0846244 100644
--- a/trt_utilities.py
+++ b/trt_utilities.py
@@ -248,6 +248,8 @@ class Engine:
         self.context = None
         self.buffers = OrderedDict()
         self.tensors = OrderedDict()
+        self.inputs = {}  # Initialize here
+        self.outputs = {}  # Initialize here
         self.cuda_graph_instance = None  # cuda graph
         self.graph = None