提交 f057d89a 编写于 作者: J Jeng Bai-Cheng 提交者: Zhaolong Xing

Fix Python inference typos (#1561)

There are 2 typos at `python_infer_cn.md`.
1. The correct parameter is Float32 instead of kFloat32
2. The correct argument is precision_mode instead of precision
上级 775a5c47
......@@ -87,7 +87,7 @@ config.enable_use_gpu(100, 0) # 初始化100M显存,使用gpu id为0
config.gpu_device_id() # 返回正在使用的gpu id
config.disable_gpu() # 禁用gpu
config.switch_ir_optim(True) # 开启IR优化
config.enable_tensorrt_engine(precision=AnalysisConfig.Precision.kFloat32,
config.enable_tensorrt_engine(precision_mode=AnalysisConfig.Precision.Float32,
use_calib_mode=True) # 开启TensorRT预测,精度为fp32,开启int8离线量化
config.enable_mkldnn() # 开启MKLDNN
```
......@@ -135,7 +135,7 @@ results = predictor.run([x_t, y_t])
* `enable_tensorrt_engine(workspace_size: int = 1 << 20,
max_batch_size: int,
min_subgraph_size: int,
precision: AnalysisConfig.precision,
precision_mode: AnalysisConfig.precision,
use_static: bool,
use_calib_mode: bool) -> None`
* `enable_mkldnn() -> None`
......@@ -209,4 +209,4 @@ def parse_args():
if __name__ == "__main__":
main()
```
\ No newline at end of file
```
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册