導入包。
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