zoukankan      html  css  js  c++  java
  • 深度之眼PyTorch训练营第二期---12、TensorBoard的使用

    一、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

  • 相关阅读:
    es6 语法 (数值扩展)
    手机日期插件 (转加上自己喜欢的)
    仿微信抢红包(js 转)
    默认时间为今天
    es6 语法 (解构赋值)
    es6 语法 (let 和const)
    es6环境搭建
    express 安装和运行
    git 常用操作,下拉,提交,更新,还原
    排序。
  • 原文地址:https://www.cnblogs.com/cola-1998/p/11904801.html
Copyright © 2011-2022 走看看