From c825ee75639adf39781445f7dd99342828a428cb Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Sat, 8 May 2021 10:23:49 +0800 Subject: [PATCH] set init gpu memory more small --- paddleocr.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddleocr.py b/paddleocr.py index c5da7248..b32e9fcb 100644 --- a/paddleocr.py +++ b/paddleocr.py @@ -179,7 +179,7 @@ def parse_args(mMain=True, add_help=True): parser.add_argument("--use_gpu", type=str2bool, default=True) parser.add_argument("--ir_optim", type=str2bool, default=True) parser.add_argument("--use_tensorrt", type=str2bool, default=False) - parser.add_argument("--gpu_mem", type=int, default=8000) + parser.add_argument("--gpu_mem", type=int, default=500) # params for text detector parser.add_argument("--image_dir", type=str) -- GitLab