pytorch tensorboard

TORCH.UTILS.TENSORBOARD

Before going further, more details on TensorBoard can be found at https://www.tensorflow.org/tensorboard/

Once you’ve installed TensorBoard, these utilities let you log PyTorch models and metrics into a directory for visualization within the TensorBoard UI. Scalars, images, histograms, graphs, and embedding visualizations are all supported for PyTorch models and tensors as well as Caffe2 nets and blobs.

The SummaryWriter class is your main entry to log data for consumption and visualization by TensorBoard. For example:

demo_1

# -*- encoding: utf-8 -*-
"""
@File    : demo_1.py
@Time    : 2021-08-11 10:00
@Author  : XD
@Email   : gudianpai@qq.com
@Software: PyCharm
"""
import torch
import torchvision
from torch.utils.tensorboard import SummaryWriter
from torchvision import datasets, transforms

# Writer will output to ./runs/ directory by default
writer = SummaryWriter()

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 grayscale rather than RGB
model.conv1 = torch.nn.Conv2d(1, 64, kernel_size=7, stride=2, padding=3, bias=False)
images, labels = next(iter(trainloader))

grid = torchvision.utils.make_grid(images)
writer.add_image('images', grid, 0)
writer.add_graph(model, images)
writer.close()

终端输入命令
注意＋host

(pytorch_gpu) G:\各种代码\随便写的>tensorboard --logdir=runs --host=127.0.0.1
TensorFlow installation not found - running with reduced feature set.
TensorBoard 1.14.0 at http://127.0.0.1:6006/ (Press CTRL+C to quit)

…
第一个demo跑通即可
后续demo可以看官方文档

关于类

torch.utils.tensorboard.writer.SummaryWriter(log_dir=None, comment='', purge_step=None, max_queue=10, flush_secs=120, filename_suffix='')

将条目直接写入日志目录中的事件文件，以供TensorBoard使用。

SummaryWriter类提供了一个高级API，用于在给定目录中创建事件文件，并向其中添加摘要和事件。该类异步更新文件内容。这允许训练程序调用方法，直接从训练循环向文件添加数据，而不会减慢训练速度。
创建一个SummaryWriter，将事件和摘要写入事件文件

1. log_dir:存储events文件的地址
2. 注释日志目录后缀附加到默认日志目录。如果分配了log_dir，则此参数无效。
   pass