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");