目录

基于PyTorch的深度学习6可视化工具Tensorboard

目录

基于PyTorch的深度学习6——可视化工具Tensorboard

先安装tensorflow(CPU或GPU版)​,然后安装tensorboardX,在命令行运行以下命令即可。

pip install tensorboardX

使用tensorboardX的一般步骤如下所示。

1)导入tensorboardX,实例化SummaryWriter类,指明记录日志路径等信息。

from tensorboardX import SummaryWriter
#实例化SummaryWriter,并指明日志存放路径。在当前目录没有logs目录将自动创建。
writer = SummaryWriter(log_dir='logs')
#调用实例
writer.add_xxx()
#关闭writer
writer.close()

2)调用相应的API接口,接口一般格式为:

add_xxx(tag-name, object, iteration-number)
#即add_xxx(标签,记录的对象,迭代次数)

3)启动tensorboard服务:cd到logs目录所在的同级目录,在命令行输入如下命令,logdir等式右边可以是相对路径或绝对路径。

tensorboard --logdir=logs --port 6006   
#如果是Windows环境,要注意路径解析,如
#tensorboard --logdir=r'D:\myboard\test\logs' --port 6006

4)web展示。在浏览器输入:

http://服务器IP或名称:6006  #如果是本机,服务器名称可以使用localhost