diff --git a/python/config.yaml b/python/config.yaml
index 510dab6..559a31f 100644
--- a/python/config.yaml
+++ b/python/config.yaml
@@ -13,7 +13,7 @@ Det:
     CUDAExecutionProvider:
         device_id: 0
         arena_extend_strategy: kNextPowerOfTwo
-        gpu_mem_limit: 2 * 1024 * 1024 * 1024
+        # gpu_mem_limit: 2 * 1024 * 1024 * 1024
         cudnn_conv_algo_search: EXHAUSTIVE
         do_copy_in_default_stream: true
 
@@ -46,7 +46,7 @@ Cls:
     CUDAExecutionProvider:
         device_id: 0
         arena_extend_strategy: kNextPowerOfTwo
-        gpu_mem_limit: 2 * 1024 * 1024 * 1024
+        # gpu_mem_limit: 2 * 1024 * 1024 * 1024
         cudnn_conv_algo_search: EXHAUSTIVE
         do_copy_in_default_stream: true
 
@@ -64,7 +64,7 @@ Rec:
     CUDAExecutionProvider:
         device_id: 0
         arena_extend_strategy: kNextPowerOfTwo
-        gpu_mem_limit: 2 * 1024 * 1024 * 1024
+        # gpu_mem_limit: 2 * 1024 * 1024 * 1024
         cudnn_conv_algo_search: EXHAUSTIVE
         do_copy_in_default_stream: true