提交 6292c438 编写于 作者: Y Yan Chunwei 提交者: GitHub

hide graph when no proto (#105)

上级 fb7b70d3
...@@ -195,6 +195,7 @@ if __name__ == '__main__': ...@@ -195,6 +195,7 @@ if __name__ == '__main__':
logger.error("should pass in logdir option") logger.error("should pass in logdir option")
sys.exit(-1) sys.exit(-1)
if options.model_pb:
# draw graph # draw graph
image_dir = os.path.join(options.logdir, "graphs") image_dir = os.path.join(options.logdir, "graphs")
if not os.path.isdir(image_dir): if not os.path.isdir(image_dir):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册