有没有办法从事件文件中读取会话数据（如计算时间，内存使用情况）？

Question

我正在编写可以从tensorflow事件文件中提取数据的脚本。

基本上，有很多方法可以从事件文件中获取摘要数据。

但是，似乎没有什么可以提取会话数据（例如每个层的计算时间和内存使用情况），这意味着运行时统计信息。

Answer 1

查看他的section官方文档。

总结一下，你只需要添加这两个参数：

run_options = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE)
run_metadata = tf.RunMetadata()

到您的sess.run调用记录运行时统计信息，并将run_metadata对象添加到您的编写器：train_writer.add_run_metadata(run_metadata, 'step%d' % i)。

您可以在图表选项卡中访问TensorBoard中的运行时统计信息。

[...]
else:  # Record train set summaries, and train
  if i % 100 == 99:  # Record execution stats
    run_options = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE)
    run_metadata = tf.RunMetadata()
    summary, _ = sess.run([merged, train_step],
                          feed_dict=feed_dict(True),
                          options=run_options,
                          run_metadata=run_metadata)
    train_writer.add_run_metadata(run_metadata, 'step%d' % i)
    train_writer.add_summary(summary, i)
    print('Adding run metadata for', i)