PyTorch使用tensorboard顯示網絡運行情況

PyTorch使用tensorboard可以顯示網絡運行情況,和TensorFlow的tensorboard使用很類似。均需要安裝tensorboard包。

pip install tensorboard==1.15.0

導入:

from torch.utils.tensorboard import SummaryWriter

指定log路徑:

log_dir = './run_logs'
self.writer = SummaryWriter(log_dir)

寫入到log:

self.board_write_step = 50#每50個step寫一次board_log
if (global_steps+1) % self.board_write_step == 0:
    self.writer.add_image("result", result_img, global_step=global_steps)
    self.writer.add_scalar("acc", acc, global_step=global_steps)

在shell中開啓tensorboard:

tensorboard --logdir=./run_logs

瀏覽器中打開localhost:6006
在這裏插入圖片描述
在這裏插入圖片描述
在這裏插入圖片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章