提交 f2a7e2a2 编写于 作者: K kinghuin 提交者: wuzewu

why no taskid

上级 77caafff
......@@ -131,6 +131,17 @@ if __name__ == '__main__':
# Construct transfer learning network
# Use "pooled_output" for classification tasks on an entire sentence.
# Use "sequence_output" for token-level output.
check = [inputs["task_ids"].name]
global_block = program.global_block()
for op in global_block.ops:
for input_arg in op.input_arg_names:
for ch in check:
if ch in input_arg:
print(op)
check.append(input_arg)
break
exit(0)
pooled_output = outputs["pooled_output"]
# Setup feed list for data feeder
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册