0.打开终端
conda install tensorboard
尽管存在torch.utils.tensorboard
,但是pytorch内部并没有安装包,需要自己安装
1.代码示例
代码里和tensorboard
相关的都写的了注释
SummaryWriter()
自动创建一个runs
的文件夹
里面存了一个event的文件
import torch
from torch.utils.tensorboard import SummaryWriter
# Writer will output to' ./runs/' directory by default.
writer = SummaryWriter()
x = torch.arange(-5, 5, 0.1).view(-1, 1)
y = -5 * x + 0.1 * torch.randn(x.size())
model = torch.nn.Linear(1, 1)
criterion = torch.nn.MSELoss()
optimizer = torch.optim.SGD(model.parameters(), lr=0.1)
def train_model(iter):
for epoch in range(iter):
y1 = model(x)
loss = criterion(y1, y)
# log a scalar value,使用add_scalar函数来完成
# For example, lets create a simple linear regression training,
# and log loss value using add_scalar
# 参数含义:title,y轴,x轴
writer.add_scalar("Loss/train", loss, epoch)
optimizer.zero_grad()
loss.backward()
optimizer.step()
train_model(10)
writer.flush()
writer.close()
# 在终端使用tensorboard logdir= ‘路径名’
# 把runs的文件路径给出来就行,tensorborad会自动递归的寻找带有event后缀的文件
# 然后直接打开 http://localhost:6006/