在Pytorch中简单使用tensorboard
作者:Ma Sizhou 时间:2021-02-05 11:07:57
一、tensorboard的简要介绍
TensorBoard是一个独立的包(不是pytorch中的),这个包的作用就是可视化您模型中的各种参数和结果。
下面是安装:
pip install tensorboard
安装 TensorBoard 后,这些实用程序使您可以将 PyTorch 模型和指标记录到目录中,以便在 TensorBoard UI 中进行可视化。 PyTorch 模型和张量以及 Caffe2 网络和 Blob 均支持标量,图像,直方图,图形和嵌入可视化。
SummaryWriter 类是您用来记录数据以供 TensorBoard 使用和可视化的主要入口。
看一个例子,在这个例子中,您重点关注代码中的注释部分:
import torch
import torchvision
from torchvision import datasets, transforms
# 可视化工具, SummaryWriter的作用就是,将数据以特定的格式存储到上面得到的那个日志文件夹中
from torch.utils.tensorboard import SummaryWriter
# 第一步:实例化对象。注:不写路径,则默认写入到 ./runs/ 目录
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)
# 让 ResNet 模型采用灰度而不是 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()
点击运行之后,我们就可以在文件夹下看到我们保存的数据了,然后我们就可以使用 TensorBoard 对其进行可视化,该 TensorBoard 应该可通过以下方式运行(在命令行):
tensorboard --logdir=runs
运行结果:
把上述的地址,粘贴到浏览器就可以看到可视化的结果了,如下所示:
接着看:
一个实验可以记录很多信息。 为了避免 UI 混乱和更好地将结果聚类,我们可以通过对图进行分层命名来对图进行分组。 例如,“损失/训练”和“损失/测试”将被分组在一起,而“准确性/训练”和“准确性/测试”将在 TensorBoard 界面中分别分组。
我们再看一个更简单的例子来理解上面的话:
from torch.utils.tensorboard import SummaryWriter
import numpy as np
# 第一步:实例化对象。注:不写参数默认是 ./run/ 文件夹下
writer = SummaryWriter()
for n_iter in range(100):
# 第二步:调用对象的方法,给文件夹存数据
writer.add_scalar('Loss/train', np.random.random(), n_iter)
writer.add_scalar('Loss/test', np.random.random(), n_iter)
writer.add_scalar('Accuracy/train', np.random.random(), n_iter)
writer.add_scalar('Accuracy/test', np.random.random(), n_iter)
writer.close()
点击运行(保存数据);
在命令行输入tensorboard --logdir=run(run是保存的数据的所在路径)
实验结果:
来源:https://blog.csdn.net/weixin_45901519/article/details/117735506
标签:Pytorch,tensorboard
0
投稿
猜你喜欢
用Vue封装导航栏组件
2023-07-02 16:51:46
python高手之路python处理excel文件(方法汇总)
2021-02-17 10:33:15
python实现定时提取实时日志程序
2023-03-03 22:31:24
关于mysql中的json解析函数JSON_EXTRACT
2024-01-23 18:29:41
ASP开发中有用的函数(function)集合(2)
2008-10-14 17:17:00
Python爬虫包BeautifulSoup异常处理(二)
2021-12-09 21:53:42
SQL Server索引管理的六大铁律
2009-03-25 14:05:00
Python爬虫Scrapy框架IP代理的配置与调试
2022-09-19 08:20:26
无法在发生错误时创建会话,请检查 PHP 或网站服务器日志,并正确配置 PHP 安装最快的解决办法
2023-09-03 19:49:50
Sql2005注射辅助脚本[粗糙版]
2024-01-24 18:53:47
在vue中配置不同的代理同时访问不同的后台操作
2024-05-09 15:08:23
详细说明关于Java的数据库连接(JDBC)
2024-01-18 09:00:16
antd table按表格里的日期去排序操作
2024-04-28 10:55:56
py3nvml实现GPU相关信息读取的案例分析
2022-01-11 11:00:52
python实现每天自动签到领积分的示例代码
2023-06-18 09:17:32
MySQL去重的方法整理
2024-01-27 07:04:42
参考sql2012存储过程写的统计所有用户表尺寸大小的示例
2024-01-25 05:50:18
echarts动态获取Django数据的实现示例
2022-08-26 02:29:06
Vuex的安装、搭建及案例详解
2024-05-29 22:20:22
Python 内置函数速查表一览
2021-10-30 07:43:06