fix eval run api usage in slim/distillation (#655)

a65395fd · Bai Yifan · GitHub · 94a2f9fb · a65395fd
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

slim/distillation/distill.py slim/distillation/distill.py +1 -1

未找到文件。
--- a/slim/distillation/distill.py
+++ b/slim/distillation/distill.py
@@ -338,7 +338,7 @@ def main():
                                              eval_prog)
            # eval
            results = eval_run(exe, compiled_eval_prog, eval_loader, eval_keys,
-                               eval_values, eval_cls)
+                               eval_values, eval_cls, cfg)
            resolution = None
            box_ap_stats = eval_results(results, cfg.metric, cfg.num_classes,
                                        resolution, is_bbox_normalized,