未验证 提交 84dde59e 编写于 作者: K Kaipeng Deng 提交者: GitHub

fix eval in distill_pruned_model (#1550)

上级 487e6e60
...@@ -317,7 +317,7 @@ def main(): ...@@ -317,7 +317,7 @@ def main():
os.path.join(save_dir, save_name)) os.path.join(save_dir, save_name))
# eval # eval
results = eval_run(exe, compiled_eval_prog, eval_loader, eval_keys, results = eval_run(exe, compiled_eval_prog, eval_loader, eval_keys,
eval_values, eval_cls) eval_values, eval_cls, cfg)
resolution = None resolution = None
box_ap_stats = eval_results(results, cfg.metric, cfg.num_classes, box_ap_stats = eval_results(results, cfg.metric, cfg.num_classes,
resolution, is_bbox_normalized, resolution, is_bbox_normalized,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册