tensorboardX可视化

安装

# 控制台运行
pip install tensorboardX
pip install tensorboard

使用–代码

# 导入可视化模块
from tensorboardX import SummaryWriter
writer = SummaryWriter('../result_tensorboard')

# 紧跟可视化的loss之后
writer.add_scalar('loss',loss.item(),tensorboard_ind)
tensorboard_ind += 1

• 关键就是writer.add_scalar(),其中有三个关键的参数：
  def add_scalar(self, tag, scalar_value, global_step):
  • tag就是一个字符串吧，在上面的代码中，我是每50个batch记录一次loss的值，所以这个tag就是’loss’
  • scalar_value就是这一次记录的标量了，上面记录的就是loss.item()。这个loss的变化应该会输出一个折线图的吧，这个scalar_value就是y轴的值
  • global_step其实就是折线图的x轴的值，所以每记录一个点就把tensorboard_ind加一

使用–可视化结果

• 在上面定义的目录会生成events.out.巴拉巴拉的文件，在控制台运行：

# logdir为上面定义的绝对路径
tensorboard --logdir=D:\Kaggle\result_tensorboard

• 会出现
  
• 点击上图中的蓝色字体，会弹出一个网页，这个网页就是tensorboald的可视化面板。