Fix eval_run in demo of pruning and sensitivity (#611)

be2c012b · whs · GitHub · a1ae96b2 · be2c012b · be2c012b
显示空白变更内容
内联并排

Showing with 16 addition and 4 deletion

slim/prune/prune.py slim/prune/prune.py +8 -2

slim/sensitive/sensitive.py slim/sensitive/sensitive.py +8 -2

未找到文件。
--- a/slim/prune/prune.py
+++ b/slim/prune/prune.py
@@ -293,8 +293,14 @@ def main():

            if FLAGS.eval:
                # evaluation
-                results = eval_run(exe, compiled_eval_prog, eval_loader,
-                                   eval_keys, eval_values, eval_cls)
+                results = eval_run(
+                    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

--- a/slim/sensitive/sensitive.py
+++ b/slim/sensitive/sensitive.py
@@ -131,8 +131,14 @@ def main():

        compiled_eval_prog = fluid.compiler.CompiledProgram(program)

-        results = eval_run(exe, compiled_eval_prog, eval_loader, eval_keys,
-                           eval_values, eval_cls)
+        results = eval_run(
+            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