From 6241b8f9ca0c17933d91646dc0ab4c9827760015 Mon Sep 17 00:00:00 2001 From: WenmuZhou Date: Thu, 22 Oct 2020 19:03:24 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E9=9D=99=E6=80=81=E5=9B=BE?= =?UTF-8?q?=E7=9A=84create=5Fpredictor?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tools/infer/utility.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 6e49357b..cfbec956 100755 --- a/tools/infer/utility.py +++ b/tools/infer/utility.py @@ -85,8 +85,8 @@ def create_predictor(args, mode, logger): if model_dir is None: logger.info("not find {} model file path {}".format(mode, model_dir)) sys.exit(0) - model_file_path = model_dir + "/__model__" - params_file_path = model_dir + "/__variables__" + model_file_path = model_dir + "/model" + params_file_path = model_dir + "/params" if not os.path.exists(model_file_path): logger.info("not find model file path {}".format(model_file_path)) sys.exit(0) -- GitLab