From fd4cd752659cdb65ec53b390f679e63642d68176 Mon Sep 17 00:00:00 2001 From: Alex Bezdieniezhnykh Date: Thu, 24 Apr 2025 16:24:15 +0300 Subject: [PATCH] conversion to model with batch 4 is DONE --- inference/tensorrt_engine.py | 1 - 1 file changed, 1 deletion(-) 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)