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
: