pytorch中使用tensorboard

可以用tensorboard的UI，看训练过程中的 标量Scalars, 图片images, 柱状分布图histograms, 网络图graphs。

一般安装了torch就有tensorboard,导入方式如下：

from torch.utils.tensorboard import SummaryWriter

官网中的例子：

官网中的例子我加上了路径，因为当下路径不太好找。

import torch
import torchvision
from torch.utils.tensorboard import SummaryWriter
from torchvision import datasets, transforms

log_dir=r'C:\Users\14497\Desktop\result\log_dir'
# Writer will output to ./runs/ directory by default
writer = SummaryWriter(log_dir) # 如果log_dir是空的，就是当前路径

transform = transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.5,), (0.5,))])
trainset = datasets.MNIST('mnist_train', train=True, download=True, transform=transform)
trainloader = torch.utils.data.DataLoader(trainset, batch_size=64, shuffle=True)
model = torchvision.models.resnet50(False)
# Have ResNet model take in gra