Skip to content

Commit e756b15

Browse files
fix docstring
1 parent 790b2fc commit e756b15

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

deploy/python_infer/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ class Predictor:
4242
onnx_path (Optional[str], optional): Path to the ONNX model file. Defaults to None.
4343
ir_optim (bool, optional): Whether to use IR optimization. Defaults to True.
4444
min_subgraph_size (int, optional): Minimum subgraph size for IR optimization. Defaults to 15.
45-
gpu_mem (int, optional): Maximum GPU memory(MB) to use. Defaults to 500(MB).
45+
gpu_mem (int, optional): Initial size of GPU memory pool(MB). Defaults to 500(MB).
4646
gpu_id (int, optional): GPU ID to use. Defaults to 0.
4747
num_cpu_threads (int, optional): Number of CPU threads to use. Defaults to 1.
4848
"""

0 commit comments

Comments
 (0)