fix the save_in_nodes bug.

6 years ago · 6b854f3e1f
parent 183bacebe3
commit 6b854f3e1f
1 changed files with 1 additions and 1 deletions
--- a/python/paddle/fluid/contrib/slim/quantization/quantization_strategy.py
+++ b/python/paddle/fluid/contrib/slim/quantization/quantization_strategy.py
@ -158,7 +158,7 @@ class QuantizationStrategy(Strategy):
                ]

            if self.save_in_nodes == None:
-                in_vars = list(context.eval_graph.out_nodes.values())
+                in_vars = list(context.eval_graph.in_nodes.values())
            else:
                in_vars = self.save_in_nodes