安装
不啰嗦
先装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