未验证 提交 8e5bde6c 编写于 作者: L littletomatodonkey 提交者: GitHub

Update predict.py

上级 07330e0e
......@@ -39,8 +39,8 @@ def create_paddle_predictor(args):
config.switch_ir_optim(args.ir_optim) # default true
if args.use_tensorrt:
config.enable_tensorrt_engine(
precision_mode=Config.PrecisionType.Half
if args.use_fp16 else Config.PrecisionType.Float32,
precision_mode=Config.Precision.Half
if args.use_fp16 else Config.Precision.Float32,
max_batch_size=args.batch_size)
config.enable_memory_optim()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册