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