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:
|
||||
option.use_trt_backend()
|
||||
option.set_trt_input_shape("inputs", [1, 3, 224, 224],
|
||||
[1, 3, 224, 224], [1, 3, 224, 224])
|
||||
return option
|
||||
|
||||
|
||||
|
Reference in New Issue
Block a user