TensorboardX的使用【有手就⭐系列】

可以理解为在PyTorch中使用TensorFlow中的可视化工具Tensorboard

• Github
• 文档

0. 安装 pip install tensorboardX

1. 实例化一个SummaryWriter(记录器)对象, 用于后续保存标量(scalar)/图像(image)/图(graph)等日志文件

   from tensorboardX import SummaryWriter #SummaryWriter Encapsultes everything      
   log_dir = "" 
   writer = SummaryWriter(log_dir) #实例化对象时指定存放log的目录
   

2. 保存sth(something)
   通用的API格式：add_sth(tag_name, object, iter_num)
   举例说明：保存标量 writer.add_scalar('loss', value, iteration)

3. 可视化网络 Add graph
   https://tensorboardx.readthedocs.io/en/latest/tutorial.html#add-graph

4. 监测训练过程
   tensorboard --logdir your_log_dir 注意是日志目录，而不是要指定日志文件。tensorboard命令可以使用Tab补全

tensorboardX的使用

https://tensorboardx.readthedocs.io/en/latest/tutorial.html#