Pytorch – Pytoch结合Tensorboard实现数据可视化
本文作者:StubbornHuang
版权声明:本文为站长原创文章,如果转载请注明原文链接!
原文标题:Pytorch – Pytoch结合Tensorboard实现数据可视化
原文链接:https://www.stubbornhuang.com/2591/
发布于:2023年04月18日 10:06:24
修改于:2023年04月18日 10:08:30
1 Tensorboard
1.1 安装和卸载
使用以下命令安装
pip install tensorboard
卸载命令
pip uninstall tensorboard
1.2 Tensorboard的工作逻辑
Tensorboard的工作逻辑其实很简单,首先将你所关心的数据保存到你所指定的文件夹中,这一步在你构造SummaryWriter对象的时候指定,然后再读取这个文件夹的数据使用浏览器显示出来。
2 在Pytorch中使用Tensorboard
首先需要在使用tensorboard的地方导入tensorboard
from torch.utils.tensorboard import SummaryWriter
然后实例化SummaryWrite
对象
summary_writer = SummaryWriter('./path/to/log')
这里传递的路径就是需要保存数据集的文件夹,在SummaryWrite
对象实例化完成之后就可以使用这个对象写入数据。
2.1 写入数据
SummaryWrite对象包含很多方法,其中写入数值就是我们经常使用的方法,其对应的函数原型为
writer.add_scalar(tag, scalar_value, global_step=None, walltime=None)
上述的tag为指定可视化时这个变量的名字,scalar_value是你要存的值,global_step可以理解为x轴坐标。
一个简单的示例
from torch.utils.tensorboard import SummaryWriter
for epoch in range(100)
train_loss = model(input)
writer.add_scalar('loss', loss, epoch)
另外有一个细节的问题就是,我们可以用以下的方式
writer.add_scalar('Loss/train', train_loss, epoch)
writer.add_scalar('Loss/test', test_loss, epoch)
来实现变量的归类,通过这种方式就可将train
和test
这两个loss显示在Loss
的同一个section,如下图所示
2.2 在网页中查看可视化结果
如果按照上述方式通过SummaryWrite对象写入了数据,然后就可以通过网页对写入的数据进行可视化了,我们可以在命令行中使用以下命令
tensorboard --logdir=./path/to/the/folder --port 6666
这里的logdir
就是数据保存的文件夹路径,port
为本地浏览器端口,然后我们就可以在浏览器中打开localhost:6666这个网址来查看数据可视化网页,这里需要保证的是所指定的端口并没有被其他程序占用。
当前分类随机文章推荐
- Pytorch - 使用torchsummary/torchsummaryX/torchinfo库打印模型结构、输出维度和参数信息 阅读1521次,点赞1次
- Pytorch - 多GPU训练方式nn.DataParallel与nn.parallel.DistributedDataParallel的区别 阅读886次,点赞0次
- Pytorch – 使用torch.matmul()替换torch.einsum('bhxyd,md->bhxym',(a,b))算子模式 阅读1079次,点赞0次
- Pytorch - torch.nn.Conv1d参数详解与使用 阅读2414次,点赞0次
- Pytorch – 使用torch.matmul()替换torch.einsum(‘nkctv,kvw->nctw’,(a,b))算子模式 阅读1073次,点赞0次
- Pytorch - masked_fill方法参数详解与使用 阅读747次,点赞0次
- Pytorch - 内置的CTC损失函数torch.nn.CTCLoss参数详解与使用示例 阅读1247次,点赞1次
- 深度学习 - 我的深度学习项目代码文件组织结构 阅读1354次,点赞3次
- Pytorch - 模型微调时删除原有模型中的某一层的方法 阅读2038次,点赞0次
- Pytorch - torch.nn.Conv2d参数详解与使用 阅读465次,点赞0次
全站随机文章推荐
- 资源分享 - Div, Grad, Curl, and All That - An Informal Text on Vector Calculus , Third Edition 英文高清PDF下载 阅读1425次,点赞0次
- 资源分享 - Mathematical Basics of Motion and Deformation in Computer Graphics , Second Edition 英文PDF下载 阅读585次,点赞0次
- 资源分享 - Graphics Gems III 英文高清PDF下载 阅读2131次,点赞0次
- WordPress - WordPress后台登录设置验证码,防止恶意爆破网站 阅读3908次,点赞0次
- C++STL容器 - std::map删除指定元素 阅读2051次,点赞0次
- 资源分享 - GPU Pro 360 - Guide to Lighting 英文高清PDF下载 阅读2644次,点赞0次
- C++ - 使用cuda api获取本机显卡数量和显卡信息 阅读160次,点赞0次
- 资源分享 - 计算机动画算法与技术,Computer Animation - Algorithms and Techniques(Second Edition)中文版PDF下载 阅读1343次,点赞0次
- NCNN - Windows下使用Visual Studio编译NCNN小白教程 阅读209次,点赞0次
- Python - itertools.groupby使用详解 阅读847次,点赞1次
评论
169