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