fix the save_in_nodes bug.

6b854f3e · Zhen Wang · 183baceb · 6b854f3e
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

python/paddle/fluid/contrib/slim/quantization/quantization_strategy.py .../fluid/contrib/slim/quantization/quantization_strategy.py +1 -1

未找到文件。
--- 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