首页 > 其他 > 详细

深度之眼PyTorch训练营第二期---12、TensorBoard的使用

时间:2019-11-21 14:22:02      阅读:131      评论:0      收藏:0      [点我收藏+]

一、TensorBoard

TensorBoard:TensorFlow中强大的可视化工具

技术分享图片

 

 SummaryWriter

功能:提供创建event file的高级接口

主要属性:

log_dir:event file输出文件夹

comment:不指定log_dir时,文件夹后缀

filename_suffix:event file文件名后缀

1.add_scalar()

功能:记录标量

tag:图像的标签名,图的唯一标识

scalar_step:要记录的标量

global_step:x轴

2.add_scalars()

main_tag:该图的标签

tag_scalar_dict:key是变量的tag,value是变量的值

3.add_histogram()

功能:统计直方图与多分位数折线图

tag:图像的标签名,图的唯一标识

values:要统计的参数

global_step:y轴

bins:取直方图的bins

4.add_image()

功能:记录图像

tag:图像的标签名,图的唯一标识

img_tensor:图像数据,注意尺寸

global_step:x轴

dataformats:数据形式,CHW,HWC,HW

 

torchvision.utils.make_grid

功能:制作网格图像

tensor:图像数据,B*C*H*W形式

nrow:行数(列数自动计算)

padding:图像间距(像素单位)

normalize:是否将像素值标准化

range:标准化范围

scale_each:是否单张图维度标准化

pad_value:padding的像素值

 

5.add_graph()

功能:可视化模型计算图

model:模型,必须是nn.Module

input_to_model:输出给模型的数据

verbose:是否打印计算图结构信息

 

torchsummary

功能:查看模型信息,便于调试

model:pytorch模型

input_size:模型输入size

batch_size:batch size

device:“cuda” or “cpu”

 

github:https://github.com/sksq96/pytorch-summary

深度之眼PyTorch训练营第二期---12、TensorBoard的使用

原文:https://www.cnblogs.com/cola-1998/p/11904801.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!