參考官網

導入tensorboard

from torch.utils.tensorboard import SummaryWriter

創建SummaryWriter實例
writer=SummaryWriter( log_dir=None, comment='', purge_step=None, max_queue=10, flush_secs=120, filename_suffix='')
參數log_dir生成日志文件保存的路徑，默認是./runs,一般只需設置這個，其余選默認值

SummaryWriter可以記錄scalar,model graph和images

• 記錄scalar writer.add_scalar('train/loss',loss_data,i+1)，第一個參數是表格中顯示出的變量名；第二個參數是代碼中的變量名，即要跟蹤的變量；第三個參數相當于index
• 記錄model graph(網絡模型)

input = torch.zeros(1,3,224,224).cuda()# 定義輸入圖像尺寸 writer.add_graph(model,input)

• 查看images

images,label = next(iter(train_dataloader)) grid = torchvision.utils.make_grid(images) writer.add_image('images', grid, 0)

最后記得writer.cloae()

在終端輸入tensorboard --logdir=./runs ,點擊跳轉鏈接，可以看到下圖
logdir是保存記錄的路徑

總結

以上是生活随笔為你收集整理的pytorch中tensorboard使用的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。