diff --git a/example/post_training_quantization/pytorch_yolo_series/eval.py b/example/post_training_quantization/pytorch_yolo_series/eval.py index 6c2462acf4a19e7402990847b7363d347972f7f1..498eb5f6ea8f27d6a10f95047527f110101533cc 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(