第一步安装使用tensorboard。
1、若是torch。安装tensorboardX;tensorflow一般安装时已经装好tensorboard。
以torch为例。
在自己的程序中适当位置添加以下代码即可。
from tensorboardX import SummaryWriter
writer = SummaryWriter('log/')
for i in range(50000):
training_loss = ....
writer.add_scalar('train/loss', training_loss, i)
运行程序…在log文件夹中出现类似events.out.tfevents.1563156145.2fb9963ad7f4的文件
第二步:
xshell中文件-》属性出现如下界面,点击隧道-》添加。设置如下:只需将两个端口设置为6006即可。
点击确定,连接。
命令行输入:
tensorboard --logdir='log'
在谷歌浏览器输入网址127.0.0.1:6006即可。