未验证 提交 da3aefad 编写于 作者: C cuicheng01 提交者: GitHub

Merge pull request #1492 from cuicheng01/release/2.3

[cherry-pick]update predictor.py
......@@ -60,6 +60,7 @@ class Predictor(object):
precision_mode=Config.Precision.Half
if args.use_fp16 else Config.Precision.Float32,
max_batch_size=args.batch_size,
workspace_size=1 << 30,
min_subgraph_size=30)
config.enable_memory_optim()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册