fix model zoo #568

Signed-off-by: YunLiu <[email protected]>
Project-MONAI · Apr 16, 2024 · bad49ec · bad49ec
1 parent bff4b15
commit bad49ec
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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)