Fix saving in quantization strategy. (#16474)

wanghaoshuang · web-flow · commit ecc3088df830 · 2019-03-28T10:49:26.000+08:00
test=develop
diff --git a/python/paddle/fluid/contrib/slim/quantization/quantization_strategy.py b/python/paddle/fluid/contrib/slim/quantization/quantization_strategy.py
@@ -152,7 +152,7 @@ def on_epoch_end(self, context):
                 ]
 
             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
 

Original file line number	Diff line number	Diff line change
`@@ -152,7 +152,7 @@ def on_epoch_end(self, context):`
`152`	`152`	`]`
`153`	`153`
`154`	`154`	`if self.save_in_nodes == None:`
`155`		`- in_vars = list(context.eval_graph.out_nodes.values())`
	`155`	`+ in_vars = list(context.eval_graph.in_nodes.values())`
`156`	`156`	`else:`
`157`	`157`	`in_vars = self.save_in_nodes`
`158`	`158`