一、tensorboard编码保存日志
详细信息可以参考下面链接:https://zhuanlan.zhihu.com/p/54947519
以下是本人画loss损失函数步骤:
1.定义保存数据的文件夹
from tensorboardX import SummaryWriter
write = SummaryWriter(log_dir='./checkpoints/flownet3d') #将数据存放在这个文件夹中。
当前文件夹为flownet3D,保存数据的文件夹如下:
2. 写入数据
write.add_scalar('train_loss', train_loss, epoch) #每进行完一个epoch,进行标量的可视化
#参数:tag,纵坐标值,横坐标值
二、tensorboard在windows上的使用方法
终端输入:tensorboard --logdir="checkpoints/flownet3d"
在pycharm的terminal中输入,注意当面目录位置。我这里是flownet3D目录下,logdir中的文件位置要与其对应。因为这里我们需要打开我们保存的数据记录。
运行完上面语句后会生成一个网址,点击进入即可。
三、tensorboard在服务器上的使用方法
- 连接ssh时,增加-L 指令 将服务器的6006端口重定向到自己机器上来:
远程登录服务器命令:ssh -L 16006:127.0.0.1:6006 account@server.address
代替一般ssh远程登录命令:ssh account@server.address
- 和原来一样进入需要的目录,这里是flownet3D。
- 输入:
tensorboard --logdir="checkpoints/flownet3d"
- 最后,在本地访问地址:http://127.0.0.1:16006/
效果如下: