fix the save_in_nodes bug.

wzzju · wzzju · commit 6b854f3e1f44 · 2019-03-27T17:19:10.000+08:00
diff --git a/python/paddle/fluid/contrib/slim/quantization/quantization_strategy.py b/python/paddle/fluid/contrib/slim/quantization/quantization_strategy.py
@@ -158,7 +158,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
`@@ -158,7 +158,7 @@ def on_epoch_end(self, context):`
`158`	`158`	`]`
`159`	`159`
`160`	`160`	`if self.save_in_nodes == None:`
`161`		`- in_vars = list(context.eval_graph.out_nodes.values())`
	`161`	`+ in_vars = list(context.eval_graph.in_nodes.values())`
`162`	`162`	`else:`
`163`	`163`	`in_vars = self.save_in_nodes`
`164`	`164`