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

 


免責聲明!

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



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