tensorboard基本介绍
conda下安装tensorboard
使用conda安装
bash
conda install tensorboard
2、tensorboard
的基本使用
1、创建一个tensorboard-test.py
代码如下
py
# tensorboard-test.py
from tensorboardX import SummaryWriter
# 创建一个"my-log"的tensorboard日志
writer = SummaryWriter("my-log")
for i in range(100):
writer.add_scalar("a", i, global_step=i)
writer.add_scalar("b",i**2, global_step=i)
writer.close()
以上代码,执行后,将在执行目录下,创建一个
my-log
的文件夹
2、展示日志
打开命令终端,
bash
cd my-log
tensorboard --logdir ./
浏览器输入http://loclahost:6006
默认使用6006端口,如需要修改端口启动可使用
tensorboard --logdir <dir-log> --port <port-num>
tensorboard
集成在jupyter-notebook
打开jupyter-notebook
bash
$ jupyter-notebook
输入如下
py
%load_ext tensorboard
%tensorboard --port {PORT} --logdir {BASE_DIR}
完整代码参考如下