tensorboard基本使用

tensorboard基本介绍

conda下安装tensorboard

使用conda安装

conda install tensorboard

2、tensorboard的基本使用

1、创建一个tensorboard-test.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、展示日志

打开命令终端，

cd my-log
tensorboard --logdir ./

浏览器输入http://loclahost:6006

默认使用6006端口，如需要修改端口启动可使用 tensorboard --logdir <dir-log> --port <port-num>

tensorboard集成在jupyter-notebook

打开jupyter-notebook

$ jupyter-notebook

输入如下

%load_ext tensorboard
%tensorboard  --port {PORT} --logdir {BASE_DIR}

完整代码参考如下