PyTorch笔记21--TensorBoard使用

三天没吃小孩了

已于 2024-02-21 16:53:05 修改

阅读量358

点赞数 7

分类专栏： PyTorch学习文章标签： pytorch 笔记人工智能

于 2024-02-21 16:47:26 首次发布

本文链接：https://blog.csdn.net/zjhzxsljl/article/details/136215471

版权

PyTorch学习专栏收录该内容

32 篇文章 0 订阅

订阅专栏

SummaryWriter是一个高级接口，用于PyTorch项目中记录标量、图像、直方图等数据，以及可视化模型计算图。它支持添加单个或多个标签的值，有助于性能监控和模型调试。

摘要由CSDN通过智能技术生成

SummaryWriter

class SummaryWriter(object):#提供创建event file的高级接口
    def __init__(self, 
                log_dir=None, #event file输出文件夹
                comment='', #不指定log_dir时，文件夹后缀
                purge_step=None, 
                max_queue=10, 
                flush_secs=120, 
                filename_suffix=''#event file文件名后缀
                )

add_scalar()

add_scalar(#记录标量
    tag, #图像的标签名，图的唯一标识
    scalar_value, #要记录的标量
    global_step=None, #x轴
    walltime=None
)

add_scalars()

add_scalars(
    main_tag, #该图的标签
    tag_scalar_dict, #key是变量的tag，value是变量的值
    global_step=None, 
    walltime=None
)

add_histogram()

add_histogram(#统计直方图与多分位数折线图
    tag, #图像的标签名，图的唯一标识
    values, #要统计的参数
    global_step=None, #y轴
    bins='tensorflow', #取直方图的bins
    walltime=None
)

add_image()

add_image(#记录图像
    tag, #图像的标签名，图的唯一标识
    img_tensor, #图像数据，注意尺度
    global_step=None, #x轴
    walltime=None, 
    dataformats='CHW'#数据形式，CHW，HWC，HW
)

torchvision.utils.make_grid

make_grid(#制作网格图像
    tensor, #图像数据, B*C*H*W形式
    nrow=8, #行数（列数自动计算）
    padding=2, #图像间距（像素单位）
    normalize=False, #是否将像素值标准化
    range=None, #标准化范围
    scale_each=False, #是否单张图维度标准化
    pad_value=0#padding的像素值
)

add_graph()

add_graph(#可视化模型计算图
    model, #模型，必须是 nn.Module
    input_to_model=None, #输出给模型的数据
    verbose=False#是否打印计算图结构信息
)

torchsummary

summary(#查看模型信息，便于调试
    model, #pytorch模型
    input_size, #模型输入size
    batch_size=-1, #batch size
    device="cuda"#“cuda” or “cpu”
)
#github: https://github.com/sksq96/pytorch-summary