tensorboard及tensorboardX基本介绍
tensorboard是谷歌开发的一款可用于tensorflow训练数据监控的可视化图形界面。并包含理解,调试和优化模型的其他工具。
而pytorch后面也放弃了自研的可视化组件,转而也支持了改可视化工具,可直接通过 from torch.utils.tensorboard import SummaryWriter
使用,前提是需要先安装好tensorboard
,如何安装见下文;
而tensorboardX
则社区版的基于tensorboard
但是,却可以脱离tensorflow
而独立使用的组件(之前的tensorboard是需要基于tensorflow的);
tensorboardX
安装后,可以正常import进入项目,并记录日志;但在真正查看tensorboardX输出的数据时,需要使用tensorboard --logdir <dir>
命令,该命令的使用是需要安装tensorboard
的
conda下安装tensorboard
使用conda安装
conda install tensorboard
conda install tensorboardx
2、tensorboard
的基本使用
1、创建一个tensorboard-test.py