使用TensorBoard的基本步驟如下:
建立TensorFlow計算圖(graph)並定義summary operations來記錄數據,如使用tf.summary.scalar記錄標量數據、使用tf.summary.histogram記錄直方圖、使用tf.summary.distribution記錄數據分布、使用tf.summary.image記錄圖像數據等。
使用tf.summary.merge_all()合併所有summary operations,以便一次性運行所有記錄節點。
使用tf.summary.FileWriter將運行後的數據保存到本地磁碟。
在訓練程式結束後,打開命令行,輸入tensorboard --logdir=
在瀏覽器中打開http://localhost:6006(或相應地址)查看可視化的結果,這裡可以查看訓練過程中指標的變化、模型結構、模型權重直方圖等信息。
此外,在Jupyter Notebooks中使用TensorBoard,可以運行%load_ext tensorboard來載入TensorBoard擴展,並使用%tensorboard --logdir logs來啟動TensorBoard。在Google Colab中使用TensorBoard也類似,只需在notebook中運行相應的命令即可。