Skip to content

Commit 5ef91b4

Browse files
authored
Update eval.py
1 parent 12fc560 commit 5ef91b4

File tree

1 file changed

+2
-2
lines changed
  • example/post_training_quantization/pytorch_yolo_series

1 file changed

+2
-2
lines changed

example/post_training_quantization/pytorch_yolo_series/eval.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -51,8 +51,8 @@ def eval():
5151
val_program, feed_target_names, fetch_targets = load_inference_model(
5252
config["model_dir"].rstrip('/'),
5353
exe,
54-
model_filename=config["model_filename"],
55-
params_filename=config["params_filename"])
54+
model_filename="model.pdmodel",
55+
params_filename="model.pdiparams")
5656

5757
bboxes_list, bbox_nums_list, image_id_list = [], [], []
5858
with tqdm(

0 commit comments

Comments
 (0)