[1] 可视化界面
先上效果图:
[2] 使用教程
[2.1] 安装pytorch和tensorboard
输入命令:
pip install torch
pip install tensorboard
[2.2] 在代码中使用write
代码demo:
import numpy as np
from torch.utils.tensorboard import SummaryWriter # 也可以使用 tensorboardX
# from tensorboardX import SummaryWriter # 也可以使用 pytorch 集成的 tensorboard
writer = SummaryWriter('log') # 配置生成的数据保存的地址
for epoch in range(100):
writer.add_scalar('test/squared', np.square(epoch), epoch)
writer.close()
执行上述代码后在本文件更目录下生成一个logs文件,且包含了一个事件文件。
[2.3] 打开tensorboard面板
在pycharm中terminal终端输入:
tensorboard --logdir=logs
一定要注意起初配置的生成文件保存地址,你在terminal终端中命令的地址要能够访问的到!!!
输入命令后,会生成一个地址,访问即可。
注意尽量要用谷歌浏览器访问。我用QQ浏览器(谷歌内核)有的数据都刷不出来。