From bad49ec3677ce5a9ed01b29657c20d0b1bd1218c Mon Sep 17 00:00:00 2001 From: YunLiu <55491388+KumoLiu@users.noreply.github.com> Date: Tue, 16 Apr 2024 15:56:34 +0800 Subject: [PATCH] fix model zoo #568 Signed-off-by: YunLiu <55491388+KumoLiu@users.noreply.github.com> --- monai/networks/utils.py | 1 - 1 file changed, 1 deletion(-) diff --git a/monai/networks/utils.py b/monai/networks/utils.py index 4e6699f16b..7e3d2e2170 100644 --- a/monai/networks/utils.py +++ b/monai/networks/utils.py @@ -840,7 +840,6 @@ def _onnx_trt_compile( # set up the conversion configuration config = builder.create_builder_config() - config.set_memory_pool_limit(trt.MemoryPoolType.WORKSPACE, 1 << 31) config.add_optimization_profile(profile) if precision == "fp16": config.set_flag(trt.BuilderFlag.FP16)