We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 790b2fc commit e756b15Copy full SHA for e756b15
deploy/python_infer/base.py
@@ -42,7 +42,7 @@ class Predictor:
42
onnx_path (Optional[str], optional): Path to the ONNX model file. Defaults to None.
43
ir_optim (bool, optional): Whether to use IR optimization. Defaults to True.
44
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).
+ gpu_mem (int, optional): Initial size of GPU memory pool(MB). Defaults to 500(MB).
46
gpu_id (int, optional): GPU ID to use. Defaults to 0.
47
num_cpu_threads (int, optional): Number of CPU threads to use. Defaults to 1.
48
"""
0 commit comments