快速上手-在pytorch使用tensorboardX可视化

安装

不啰嗦
先装tensorflow，然后安装同样版本的tensorboard
pip install tensorflow2.2.0
pip install tensorboard2.2.0

导入使用tensorboardX

这里用最简单最常用的summary举例，其他的看官方文档就行了

from tensorboardX import SummaryWriter

然后实例化

writer = SummaryWriter() #不填默认在当前目录下建立runs文件，自己填就是指定存储位置

然后训练过程中记录要的数值

for i in epochs:
    ..............................
	writer.add_scalar('test_acc',acc,i)
	writer.add_scalar('avg_loss',loss,i)

最后进tensorboard看就行了
tensorboard --logdir runs