tensorboard學習筆記


TensorBoard 默認是不會記錄每個節點的用時、耗費的內存大小等這些信息的,那么如何才能在圖上顯示這些信息呢?關鍵就是如下這些代碼,主要就是在 sess.run() 中加入 options 和 run_metadata 參數。

#定義TensorFlow運行選項,設置trace_lever FULL_TRACE
run_options = tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE)
#定義TensorFlow運行元信息,記錄訓練運算時間和內存占用等信息
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%03d' % i)
train_writer.add_summary(summary, i)

tensorboard圖表解釋,https://blog.csdn.net/u010099080/article/details/77426577


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM