深度学习pytorch使用tensorboard可视化实验数据

Posted 2022-12-01 征途黯然.

[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浏览器（谷歌内核）有的数据都刷不出来。