Pytorch中Tensorboard使用方法

自己做个笔记，为了防止自己忘了

安装

pip install tensorboard

使用与启动

为了使用 TensorboardX，我们首先需要创建一个 SummaryWriter 的实例，然后再使用add_scalar方法或add_image方法，将数字或图片记录到 SummaryWriter 实例中。SummaryWriter 类的定义

torch.utils.tensorboard.writer.SummaryWriter(log_dir=None)

其中的 log_dir 表示保存日志的路径，默认会保存在“runs/ 当前时间 _ 主机名”文件夹中。

添加数据，我们来看add_scalar方法，这个方法用来记录数字常量

add_scalar(tag, scalar_value, global_step=None, walltime=None)

参数：

tag：字符串类型，表示数据的名称，不同名称的数据会使用不同曲线展示；

scalar_value：浮点型，表示要保存的数值；

global_step：整型，表示训练的 step 数；

walltime：浮点型，表示记录发生的时间，默认为 time.time()。

记录单个图像

add_image方法用来记录单个图像数据（需要 Pillow 库的支持ÿ