- tensorboard 简介
TensorBoard是TensorFlow下的一个可视化的工具,能在训练大规模神经网络时将复杂的运算过程可视化。TensorBoard能展示你训练过程中绘制的图像、网络结构等
- 在 pycharm 命令行终端输入 :pip install tensorboard 即可
- 新建python文件,测试代码如下:
from torch.utils.tensorboard import SummaryWriter
writer = SummaryWriter("logs")
for i in range(100):
writer.add_scalar("y=2x", 2*i, i)
writer.close()
- 运行上述代码后,会在代码树中生成一个“logs”文件夹:
- 在pycharm 终端中输入:tensorboard --logdir=logs 查看代码:
- 点击 http://localhost:6006/即可用浏览器打开,显示如下: