diff --git a/python/paddle/fluid/contrib/slim/quantization/quantization_strategy.py b/python/paddle/fluid/contrib/slim/quantization/quantization_strategy.py
index 6812b4c633d5b55d84fff935b696297f30b18c6b..c4b02166abf020ed561a989fa97b28c9fd44223e 100644
--- a/python/paddle/fluid/contrib/slim/quantization/quantization_strategy.py
+++ b/python/paddle/fluid/contrib/slim/quantization/quantization_strategy.py
@@ -152,7 +152,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