diff --git a/tools/infer/utility.py b/tools/infer/utility.py index 6e49357b0dff1b6274eb94e9ecac6337daef963c..cfbec95605e261994555392bedb9f3abd26c31fe 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)