diff --git a/inference/tensorrt_engine.py b/inference/tensorrt_engine.py
index b75d303..912304a 100644
--- a/inference/tensorrt_engine.py
+++ b/inference/tensorrt_engine.py
@@ -12,7 +12,6 @@ from inference.onnx_engine import InferenceEngine
 import pycuda.autoinit
 import pynvml
 
-# TODO: 2. Convert onnx model with 4 batch and make sure it is working
 
 class TensorRTEngine(InferenceEngine):
     TRT_LOGGER = trt.Logger(trt.Logger.WARNING)