我有一个在TensorFlow中实现的网络需要很长时间才能进行训练,因此需要对其进行分析以查看哪些部分会导致长运行时间.

为此,我按照指令here捕获运行时和内存信息.我的代码看起来像这样：

// define network
loss = ...
train_op = tf.train.AdamOptimizer().minimize(loss, global_step=global_step)

// run forward and backward prop for one batch
run_metadata = tf.RunMetadata()
options = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE)
_,loss,sum = sess.run([train_op,loss,sum], feed_dict=fd, options=options, run_metadata=run_metadata)
writer.add_run_metadata(run_metadata, 'step_%d' % step)

然后我可以在TensorBoard中看到“会话运行”.但是,只要加载会话运行,我的图形中的大多数操作都会变为橙色,如下所示,并且没有可用的运行时或内存信息：

根据传说,这些操作是“未使用的”.但事实并非如此,因为除了“损失”和“选择”之外的几乎所有事情都是这样的.显然,必须使用整个网络来计算损失.所以我真的不明白为什么图表会像这样显示出来.

我在特斯拉K40c上使用TF 1.3.

最佳答案 我曾经遇到过与Tensorboard相同的问题,除了渐变和优化器操作之外,我在会话运行中没有注册任何东西.

我通过将我的Tensorflow版本升级到1.4版本来修复它.