Pytorch 中的 SummaryWriter()

1 导入库函数

from torch.utils.tensorboard.writer import SummaryWriter

2 SummaryWriter(log_dir, flush_secs)

• log_dir: tensorboard 文件的存放路径

• flush_secs: 写入tensorboard 文件的时间间隔 (不是很理解)

3 示例

add_scalar(标题, y, x)

from torch.utils.tensorboard.writer import SummaryWriter

# 注意存地址的是斜杠 /
writer = SummaryWriter(log_dir='C:/Users/Zhang/Pytorch/各种函数/aboutsummarywriter')

for i in range(100):
    writer.add_scalar('y = 4x', 4 * i, i)
for i in range(100):
    writer.add_scalar('y = (1/3)x', (1 / 3) * i, i)

writer.close()

4 可视化展示及其注意事项

在Pycharm 的 terminal 中,输入以下指令

tensorboard --logdir=C:/Users/Zhang/Pytorch/各种函数/aboutsummarywriter --port=8888

• logdir 后面的等号不能加空格, 否则会报错

tensorboard: error: argument {serve,dev}: invalid choice: 'C:/Users/Zhang/Pytorch/各种函数/aboutsummarywriter' (choose from 'serve', 'dev')

• logdir 是event 文件存放的路径, 可以是绝对路径也以是相对路径