1.TensorBoard神经网络可视化工具
TensorBoard是一个强大的可视化工具,在pytorch中有两种调用方法:
1.from tensorboardX import SummaryWriter
这种方法是在官方还不支持tensorboard时网上有大神写的
2.from torch.utils.tensorboard import SummaryWriter
这种方法是后来更新官方加入的
1.1 调用方法
1.1.1 创建接口SummaryWriter
功能:创建接口
调用方法:
writer = SummaryWriter(“runs”)
参数:
log_dir:event file输出文件夹
comment:不指定log_dir时,文件夹后缀
filename_suffix:event file文件名后缀
1.1.2 记录标量add_scalars()
功能:记录标量add_scalars()
调用方法:
writer.add_scalars(“name”,{“dic”:val},epoch)
参数:
tag:图像的标签名
scalar_step:要记录的标量
global_step:轮次
(https://www.jb51.net/article/212027.htm)