Tensorboard的使用

云岛夜川川發表於2024-06-18

什麼是Tensorboard

在訓練過程中我們能希望看到訓練的過程怎麼樣,這時候就可以使用這個模組來實現。他可以顯示函式之間的變化過程。

Tensorboard的使用

Summarywriter類

這個類的作用是建立訓練日誌檔案。
Summarywriter(“檔案地址”)就可以在目錄下生成訓練的日誌,這個日誌可以用來展示成視覺化的board
通常使用的是add_scalar()生成繪製的日誌。
包含三個引數,tag,y軸,x軸

from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter('logs')

#比如繪製一個y=x*x的訓練過程

for x in range(10):

    writer.add_scalar("y=x*x",x*x,x)
writer.close()

執行之後就會在這裡面生成一個logs的日誌資料夾和內容
image

繪製board

執行程式碼後會生成日誌檔案,在命令列使用命令tensorboard --logdir=目錄 --port=埠埠預設是6006.
image

如果生成的標籤是相同的,再次繪製時就會出現亂圖。就是會增加變化過程。建議就是刪除所有的日誌檔案再次執行或者更改標籤。
image

相關文章