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