mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-10-05 16:48:03 +08:00
Update infer.py
This commit is contained in:
@@ -34,8 +34,6 @@ def build_option(args):
|
|||||||
|
|
||||||
if args.use_trt:
|
if args.use_trt:
|
||||||
option.use_trt_backend()
|
option.use_trt_backend()
|
||||||
option.set_trt_input_shape("inputs", [1, 3, 224, 224],
|
|
||||||
[1, 3, 224, 224], [1, 3, 224, 224])
|
|
||||||
return option
|
return option
|
||||||
|
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user