2017-07-15 97 views
0

我想獲取有關我的圖表需要多少內存的信息,所以我試圖使用tf.RunMetadata列車選項來檢查張量板中不同圖形組件的字節信息。我的代碼的培訓部分看起來像這樣:缺少張量節點內存信息

sess=tf.Session 
... 

for itr in xrange(MAX_STEPS): 
    train_images, train_annotations = train_dataset_reader.next_batch(BATCH_SIZE) 
    feed_dict = {x: train_images, y: train_annotations} 
    run_options = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE) 
    run_metadata = tf.RunMetadata() 
    sess.run(train_step, feed_dict=feed_dict, options=run_options, run_metadata=run_metadata) 

    #Tensorboard summary tester 
    if itr % 1000 == 0: 
     s = sess.run(merged_summary, feed_dict=feed_dict,options=run_options, run_metadata=run_metadata) 

     writer.add_summary(s, itr) 
     writer.add_run_metadata(run_metadata, 'sted%d' % itr) 

但是我仍然沒有看到我的張量板圖上的任何內存統計信息。輸出如下:enter image description here

回答

2

我剛纔有同樣的問題。內存信息僅與特定的「會話運行」一起顯示。所以你的問題沒有會話運行。

Session runs image

在我的情況下,再次運行刪除以前的logdata後,就得到了解決。