From a8c61733477fca5fccf75375292c15e7b315b3bd Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Mon, 19 Sep 2022 13:33:47 +0800 Subject: [PATCH] add txt --- tools/infer/utility.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 5177236a..793ff28a 100644 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -228,7 +228,7 @@ def create_predictor(args, mode, logger): # collect shape model_name = os.path.basename(model_dir[:-1]) if model_dir.endswith("/") else os.path.basename(model_dir) - trt_shape_f = f"{mode}_{model_name}" + trt_shape_f = f"{mode}_{model_name}.txt" if trt_shape_f is not None: if not os.path.exists(trt_shape_f): config.collect_shape_range_info(trt_shape_f) -- GitLab