From d7ac87f37d6e9161a20376a1485400520236c3cd Mon Sep 17 00:00:00 2001 From: try112 <108114971+2630879836@users.noreply.github.com> Date: Wed, 18 Jun 2025 23:37:02 +0800 Subject: [PATCH] docs: correct gpu_mem parameter description (#15773) --- deploy/cpp_infer/src/args.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deploy/cpp_infer/src/args.cpp b/deploy/cpp_infer/src/args.cpp index f1d3ae2e1e..9e1573c193 100644 --- a/deploy/cpp_infer/src/args.cpp +++ b/deploy/cpp_infer/src/args.cpp @@ -18,7 +18,7 @@ DEFINE_bool(use_gpu, false, "Inferring with GPU or CPU."); DEFINE_bool(use_tensorrt, false, "Whether use tensorrt."); DEFINE_int32(gpu_id, 0, "Device id of GPU to execute."); -DEFINE_int32(gpu_mem, 4000, "GPU id when inferring with GPU."); +DEFINE_int32(gpu_mem, 4000, "GPU memory size (MB) to use."); DEFINE_int32(cpu_threads, 10, "Num of threads with CPU."); DEFINE_bool(enable_mkldnn, false, "Whether use mkldnn with CPU."); DEFINE_string(precision, "fp32", "Precision be one of fp32/fp16/int8");