From ecc3088df830f8574cef7d4f859d93946e93be5c Mon Sep 17 00:00:00 2001 From: whs Date: Thu, 28 Mar 2019 10:49:26 +0800 Subject: [PATCH] Fix saving in quantization strategy. (#16474) test=develop --- .../fluid/contrib/slim/quantization/quantization_strategy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/contrib/slim/quantization/quantization_strategy.py b/python/paddle/fluid/contrib/slim/quantization/quantization_strategy.py index 6812b4c633d..c4b02166abf 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 -- GitLab