Pytorch 配置 Tensorboard 可视化学习(一)

安装

pip install -i https://mirrors.aliyun.com/pypi/simple/ tensorboard

使用案例

新建一个python的文件，写入：

from torch.utils.tensorboard import SummaryWriter
import time
import random

writer = SummaryWriter(log_dir='./' + time.strftime('%y-%m-%d_%H.%M', time.localtime())) # 确定路径
for i in range(10):
    # 直接写入就行，更多用法请参考后面的链接，这里就是最最最简单的例子
    writer.add_scalar('test_value', random.random(), i)
# 想写的都写完后调用这个函数
writer.close()

然后进入到目录中，运行ls可以看到创建的目录21-07-18_10.41（这个是与大家自己的时间相同哈）即可，然后运行：

tensorboard --logdir=21-07-18_10.41

得到如下界面：

参考文章

https://pytorch.org/docs/stable/tensorboard.html