From 6b854f3e1f4412b5726197bc336754f163148cd8 Mon Sep 17 00:00:00 2001 From: Zhen Wang Date: Wed, 27 Mar 2019 17:19:10 +0800 Subject: [PATCH] fix the save_in_nodes bug. --- .../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 da3510de3..aa5089112 100644 --- 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 -- GitLab