pytorch tensorboard可视化

1，安装

conda install torch tensorboard

2，代码

import torch
import os

logs = "logs"

# 日志
if not os.path.exists(logs):
    os.mkdir(logs)

from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter(log_dir="logs") #定义writer,使用默认路径


for epoch in range(100):
    writer.add_scalar("train/loss", epoch, epoch)  # 图名，值，step

writer.close()

3，运行

tensorboard --logdir=logs --port=6006  # --logdir=logs地址

4，浏览器查看地址

http://localhost:6006/