提交 c5d19022 编写于 作者: Y Yu Yang

Make helper shows the correct output value

Change-Id: Ia5358f88129b5c2e2a2653b1d85c168b80d797b3
上级 76a668cb
...@@ -1121,12 +1121,13 @@ def outputs(layers): ...@@ -1121,12 +1121,13 @@ def outputs(layers):
logger.info( logger.info(
"".join(["The input order is [", ", ".join(final_inputs), "]"]) "".join(["The input order is [", ", ".join(final_inputs), "]"])
) )
if len(final_outputs) == 0:
final_outputs = map(lambda x: x.name, layers)
logger.info( logger.info(
"".join(["The output order is [", ", ".join(final_outputs), "]" "".join(["The output order is [", ", ".join(final_outputs), "]"
])) ]))
Inputs(*final_inputs) Inputs(*final_inputs)
if len(final_outputs) != 0: Outputs(*final_outputs)
Outputs(*final_outputs)
else:
Outputs(*map(lambda x: x.name, layers))
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册