本文介绍了如何在PyTorch中利用TensorBoard的`add_scalar()`和`add_image()`函数进行训练过程的图形和图像可视化。通过`SummaryWriter`创建日志文件，使用`add_scalar()`绘制曲线图，`add_image()`展示训练结果图像。查看这些可视化数据需运行tensorboard并指定logdir，通常在本地通过http://localhost:6006/访问。
tensorboard中基本函数的用法
1.add_scalar()画图形曲线
首先通过SummaryWriter函数获得该变量writer，并且将生成的文件放在名为logs的文件夹下，代码为：
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter("logs")
接着调用add_scalar函数生成需要的曲线图，将生成的图存于上面函数创建的logs文件夹下，其中函数的第一个参数是要生成的图的名字，第二的参数是y轴，第三个参数是x轴。例如生成100以内，y=x的曲线图代码如下：
for i in range(10
最低0.47元/天 解锁文章
扫一扫
4万+
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
