machine-learning - 当我想收集和不想收集 TensorBoard 统计信息时，如何创建单个脚本文件？

Question

我想要一个脚本，它可以收集张量板数据，也可以不收集，具体取决于我如何运行它。我知道我可以传递标志来告诉我的脚本我希望它如何运行。我什至可以在脚本中对其进行硬编码，然后手动更改脚本。

这两种解决方案都有一个更大的问题。当我想要运行或不运行摘要编写器操作时，我发现自己必须在脚本中的任何地方编写 if 语句。例如，我发现我必须这样做:

if tb_sys_arg = 'tensorboard':
    merged = tf.merge_all_summaries()

然后根据 tb_sys_arg 的值运行摘要，如下所示:

if tb_sys_arg = 'tensorboard':
    merged = tf.merge_all_summaries()
else:
    train_writer = tf.train.SummaryWriter(tensorboard_data_dump_train, sess.graph)

这对我来说真的很愚蠢。我宁愿不必这样做。这是正确的方法吗？我只是不想每次运行主脚本时收集统计信息，但我也不想有两个单独的脚本。

<小时/>

作为一个轶事，几个月前我开始使用 TensorBoard，看起来我一直在运行我的主文件，如下所示:

python main.py —logdir=/tmp/mdl_logs

以便它收集张量板数据。但意识到我认为我不需要最后一个标志来收集张量板数据。时间太久了，我现在都忘了我是否真的需要它。我一直在阅读文档和教程，但似乎我不需要最后一个标志(它只需要运行网络应用程序，如 tensorboard --logdir=path/to/log-directory ，对吧？)我一直以来都做错了吗？

Answer 1

您可以在没有“摘要”服务的情况下启动 Supervisor，因此它不会运行摘要节点，请参阅 Supervisor 文档的“启动较少的服务”部分 -- https://github.com/tensorflow/tensorflow/blob/master/tensorflow/g3doc/api_docs/python/functions_and_classes/shard6/tf.train.Supervisor.md#launching-fewer-services