未验证 提交 ecc3088d 编写于 作者: W whs 提交者: GitHub

Fix saving in quantization strategy. (#16474)

test=develop
上级 2aa18e2b
...@@ -152,7 +152,7 @@ class QuantizationStrategy(Strategy): ...@@ -152,7 +152,7 @@ class QuantizationStrategy(Strategy):
] ]
if self.save_in_nodes == None: 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: else:
in_vars = self.save_in_nodes in_vars = self.save_in_nodes
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册