Pytorch的可视化工具TensorboardX—SummaryWriter详解
TensorFlow 附加的工具Tensorboard 是一个很好用的视觉化工具。他可以记录数字,影像或者是声音资讯,对于观察类神经网路训练的过程非常有帮助。很可惜的是其他的训练框架(PyTorch, Chainer, numpy)并没有这么好用的工具。tensorboardX 的目的就是让其他tensorboard 的功能都可以轻易的被非TensorFlow 的框架使用。参见:https://github.com/lanpa/tensorboardX
torch1.10版本可调用模块:torch.utlis.tensorboard.writer,参见:
https://pytorch.org/docs/stable/_modules/torch/utils/tensorboard/writer.html#SummaryWriter.add_scalar,遗憾的是用vs code测试并未成功。
SummaryWriter主要函数
标量
- 添加标量
add_scalar(self, tag, scalar_value, global_step=None, walltime=None)
- 添加标量组
add_scalars(self, main_tag, tag_scalar_dict, global_step=None, walltime=None)