有没有办法计算TensorFlow网络中每个节点的时间成本?
我发现很难找到性能瓶颈.
编辑:Timeline
分析器非常棒(/sf/ask/17360801/).
如果要查找TF上每个操作花费的时间,可以使用运行时统计信息在tensorboard中执行此操作.您将需要执行以下操作(请查看上述链接中的完整示例):
run_options = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE) run_metadata = tf.RunMetadata() sess.run(, options=run_options, run_metadata=run_metadata) your_writer.add_run_metadata(run_metadata, 'step%d' % i)
不仅仅是打印它你可以在tensorboard中看到它:
此外,单击节点将显示确切的总内存,计算时间和张量输出大小.
此外,现在tensorflow还有一个调试器.这是一个如何使用它的教程.