azaion/ai-training
1
0
Fork 0
mirror of https://github.com/azaion/ai-training.git synced 2026-04-23 02:06:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

conversion to model with batch 4 is DONE

Browse Source
This commit is contained in:
Alex Bezdieniezhnykh
2025-04-24 16:24:15 +03:00
parent 26eb3b991d
commit fd4cd75265
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
inference/tensorrt_engine.py
-1
View File
@@ -12,7 +12,6 @@ from inference.onnx_engine import InferenceEngine
import pycuda.autoinit import pycuda.autoinit
import pynvml import pynvml
# TODO: 2. Convert onnx model with 4 batch and make sure it is working
class TensorRTEngine(InferenceEngine): class TensorRTEngine(InferenceEngine):
TRT_LOGGER = trt.Logger(trt.Logger.WARNING) TRT_LOGGER = trt.Logger(trt.Logger.WARNING)