pytorch使用tensorboard

安装依赖：

sudo pip install tensorboardX

sudo pip install tensorboard

sudo pip install tensorflow

如果在安装tensorflow过程中出现cant uninstall enum34错误

可以先运行以下命令

sudo pip install --ignore-installed enum34

然后重新安装

sudo pip install tensorflow

scalar使用：

from tensorboardX import SummaryWriter

writer = SummaryWriter(log_dir='xxx')

创建一个summarywriter,指定记录文件的保存目录为xxx

writer.add_scalars('aaa/bbb', x, i)

x代表想要记录的标量（纵坐标），i代表迭代次数（横坐标）

cd到xxx上一级目录，命令行输入：

tensorboard --logdir xxx

打开输出结果中的链接，就可以在网页上看到可视化结果