SummaryWriter基本用法

SummaryWriter基本使用_summarywriter怎么用-CSDN博客

SummaryWriter是PyTorch中的一个工具，用于将模型训练过程中的日志保存到TensorBoard中。可以通过使用SummaryWriter的add_scalar()方法来保存损失函数、准确率等标量信息，使用add_histogram()方法来保存权重、梯度等张量信息，使用add_image()方法来保存图像信息等。

安装：

conda install tensorboard
或者
pip install tensorboard

导入包，调用相关方法：

import torch
import random
from torch.utils.tensorboard import SummaryWriter

writer = SummaryWriter("logs")        #当前目录下创建logs文件夹用于存放日志
for i in range(-100:100):
    writer.add_scalar("y = x*X", i*i, i)
writer.close()

运行查看结果：

tensorboard --logdir=logs           #存放日志的文件夹
或者：
tensorboard --logdir=绝对地址