2020-03-30 pytorch中的tensorboardx安装及简单使用

安装

pip install tensorflow
pip install tensorboardx
据说tensorboardx是基于tensorflow的tensorboard的所以要先安装tensorflow

简单使用

from tensorboardX import SummaryWriter   
writer = SummaryWriter()
# ...
writer.add_scalar('train_loss', loss, n)
writer.add_graph(model, torch.rand([1,3,224,224]))

可视化结果命令

tensorboard --logdir=runs
官方文档https://tensorboardx.readthedocs.io/en/latest/tutorial_zh.html