From 5ef91b47b8a6cfbfb8d088de6a04a5f6f8cc6e36 Mon Sep 17 00:00:00 2001 From: Guanghua Yu <742925032@qq.com> Date: Thu, 8 Dec 2022 19:06:59 +0800 Subject: [PATCH] Update eval.py --- .../post_training_quantization/pytorch_yolo_series/eval.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/example/post_training_quantization/pytorch_yolo_series/eval.py b/example/post_training_quantization/pytorch_yolo_series/eval.py index 6c2462ac..498eb5f6 100644 --- a/example/post_training_quantization/pytorch_yolo_series/eval.py +++ b/example/post_training_quantization/pytorch_yolo_series/eval.py @@ -51,8 +51,8 @@ def eval(): val_program, feed_target_names, fetch_targets = load_inference_model( config["model_dir"].rstrip('/'), exe, - model_filename=config["model_filename"], - params_filename=config["params_filename"]) + model_filename="model.pdmodel", + params_filename="model.pdiparams") bboxes_list, bbox_nums_list, image_id_list = [], [], [] with tqdm( -- GitLab