pytorch中tensorboard的使用

導入包。

from torch.utils.tensorboard import SummaryWriter

實例化一個summaryWriter，設置記錄保存在runs文件夾里。

writer = SummaryWriter('runs')

在運行中記錄loss：比如每20個batch記錄一次損失，logging_loss是這20次損失的總和。圖片名稱設置為“”Train/Loss“，logging_loss作為y軸，count作為x軸。

if count % 20 == 0: writer.add_scalar('Train/Loss', logging_loss / 20, count)

最后：

writer.close()

 

查看tensorboard，控制台輸入以下語句，--logdir=設置的日志文件夾。

tensorboard --logdir=runs

打開瀏覽器輸入http://localhost:6006

 

tensorborad可支持的形式有 scalar，image  figure，histogram，audio，text，graph，onnx_graph，embedding，pr_curve，video等，可以查看pytorch的文檔：https://pytorch.apachecn.org/docs/1.2/tensorboard.html