pytorch中tensorboard使用

参考官网

导入tensorboard

from torch.utils.tensorboard import SummaryWriter

创建SummaryWriter实例
writer=SummaryWriter( log_dir=None, comment='', purge_step=None, max_queue=10, flush_secs=120, filename_suffix='')
参数log_dir生成日志文件保存的路径，默认是./runs,一般只需设置这个，其余选默认值
SummaryWriter可以记录scalar,model graph和images

记录scalar writer.add_scalar('train/loss',loss_data,i+1)，第一个参数是表格中显示出的变量名；第二个参数是代码中的变量名，即要跟踪的变量；第三个参数相当于index
记录model graph(网络模型)

input = torch.zeros(1,3,224,224).cuda()# 定义输入图像尺寸
writer.add_graph(model,input)

查看images

images,label = next(iter(train_dataloader))
grid = torchvision.utils.make_grid(images)
writer.add_image('images', grid, 0)

最后记得writer.cloae()
在终端输入tensorboard --logdir=./runs ,点击跳转链接，可以看到下图
logdir是保存记录的路径

pytorch中tensorboard使用相关推荐

PyTorch中如何使用tensorboard可视化
PyTorch中如何使用tensorboard可视化调用代码: from logger
在pytorch中使用tensorboard
pytorch的tensorboard安装2.0版本,然后在pytorch的环境中安装tensoflow2.0版本. 网络开始训练之后,会生成一个文件,然后在终端运行tensorboard. 输入:t ...
pytorch中使用TensorBoard进行可视化Loss及特征图
pytorch中使用TensorBoard进行可视化Loss及特征图安装导入TensorBoard 安装TensorBoard pip install tensorboard 导入TensorBoa ...
Pytorch学习-tensorboard的使用
Pytorch学习-tensorboard的使用 1 Tensorboard简介运行机制安装及测试 2 SummaryWriter实例的使用教程 (1)初始化summaryWriter的方法 (2 ...
【深度学习】在PyTorch中构建高效的自定义数据集
文章来源于磐创AI,作者磐创AI 学习Dataset类的来龙去脉,使用干净的代码结构,同时最大限度地减少在训练期间管理大量数据的麻烦. 神经网络训练在数据管理上可能很难做到"大规模" ...
pytorch贝叶斯网络_贝叶斯神经网络：2个在TensorFlow和Pytorch中完全连接
pytorch贝叶斯网络贝叶斯神经网络 (Bayesian Neural Net) This chapter continues the series on Bayesian deep learni ...
pytorch调用tensorboard方法尝试
转载自:pytorch调用tensorboard方法尝试 tensorboard提供了用于监视训练损失很好的接口,可以帮助我们更好的调整参数.下文介绍如何在pytorch中调用tensorboard. ...
小绮的踩坑日记|小土堆pytorch视频——tensorboard
报错 "在torch.utils.tensorboard中找不到" 问题1: 在pytorch中通过pip install tensorboard正确安装tensorboard之后 ...
Pytorch求索(1)：pytorch与Tensorboard的结合使用
pytorch与tensorboard结合使用 Tensorboard Tensorboard一般都是作为tf的可视化工具,与tf深度集成,它能够展现tf的网络计算图,绘制图像生成的定量指标图以及附加 ...

pytorch中tensorboard使用

pytorch中tensorboard使用相关推荐

最新文章

热门文章