提交 5ca15816 编写于 作者: Y Yanzhan Yang 提交者: Jiaying Zhao

fix fast check (#1720)

* fix fast check

* fix style
上级 a0655313
...@@ -225,6 +225,7 @@ def save_all_op_output(feed_kv=None): ...@@ -225,6 +225,7 @@ def save_all_op_output(feed_kv=None):
fetch_names = [] fetch_names = []
for fetch in fetches: for fetch in fetches:
fetch_names.append(fetch.name) fetch_names.append(fetch.name)
feed_names = feeds
for i in range(len(ops)): for i in range(len(ops)):
op = ops[i] op = ops[i]
var_name = None var_name = None
...@@ -235,7 +236,7 @@ def save_all_op_output(feed_kv=None): ...@@ -235,7 +236,7 @@ def save_all_op_output(feed_kv=None):
if "sequence_pool" in var_name: if "sequence_pool" in var_name:
continue continue
if fast_check: if fast_check:
if var_name not in fetch_names: if var_name not in fetch_names and var_name not in feed_names:
continue continue
try: try:
data = get_var_data(var_name, feed_kv=feed_kv).flatten().tolist() data = get_var_data(var_name, feed_kv=feed_kv).flatten().tolist()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册