Skip to content

Commit

Permalink
fix docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
HydrogenSulfate committed Feb 4, 2024
1 parent 790b2fc commit e756b15
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion deploy/python_infer/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ class Predictor:
onnx_path (Optional[str], optional): Path to the ONNX model file. Defaults to None.
ir_optim (bool, optional): Whether to use IR optimization. Defaults to True.
min_subgraph_size (int, optional): Minimum subgraph size for IR optimization. Defaults to 15.
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).
gpu_id (int, optional): GPU ID to use. Defaults to 0.
num_cpu_threads (int, optional): Number of CPU threads to use. Defaults to 1.
"""
Expand Down

0 comments on commit e756b15

Please sign in to comment.