From a65395fd2758cbe8c0de638874bda2018121ecc2 Mon Sep 17 00:00:00 2001
From: Bai Yifan <me@ethanbai.com>
Date: Wed, 13 May 2020 10:57:03 +0800
Subject: [PATCH] fix eval run api usage in slim/distillation (#655)

---
 slim/distillation/distill.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/slim/distillation/distill.py b/slim/distillation/distill.py
index 70113949b..68003757d 100644
--- 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,
-- 
GitLab