Fix train_eval in quantization (#682)

3697a31e · Liufang Sang · GitHub · 42bd90d7 · 3697a31e
显示空白变更内容
内联并排

Showing with 8 addition and 2 deletion

slim/quantization/train.py slim/quantization/train.py +8 -2

未找到文件。
--- a/slim/quantization/train.py
+++ b/slim/quantization/train.py
@@ -256,8 +256,14 @@ def main():
            if FLAGS.eval:
                # evaluation
-                results = eval_run(exe, compiled_eval_prog, eval_loader,
+                results = eval_run(
-                                   eval_keys, eval_values, eval_cls)
+                    exe,
+                    compiled_eval_prog,
+                    eval_loader,
+                    eval_keys,
+                    eval_values,
+                    eval_cls,
+                    cfg=cfg)
                resolution = None
                if 'mask' in results[0]:
                    resolution = model.mask_head.resolution