From c03b0b0604f238f918b30dff3c043fb467ea27f2 Mon Sep 17 00:00:00 2001 From: zhoujun Date: Thu, 31 Dec 2020 16:31:21 +0800 Subject: [PATCH] Update utility.py --- tools/infer/utility.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 966fa3cc..c19946eb 100755 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -83,6 +83,7 @@ def parse_args(): parser.add_argument("--enable_mkldnn", type=str2bool, default=False) parser.add_argument("--use_pdserving", type=str2bool, default=False) + parser.add_argument("--use_tensorrt", type=str2bool, default=False) return parser.parse_args() -- GitLab