Pytorch Tensorboard 使用

Pytorch Tensorboard 使用

此文为 pytorch tensorboard 使用的基础总结，感觉以后会开一个专题梳理基本使用，以及一些常用函数或功能的深入分析。

导包

from tensorboardX import SummaryWriter

对象声明

writer = SummaryWriter('./tests/log/')

实例化写入

x = [9, 8, 7, 6, 5]
y = [1, 2, 3, 4, 5]
for l in range(len(x)):
    print(x[l], y[l])
    writer.add_scalar('AUC/AUC', x[l], y[l])

可视化

tensorboard --logdir='./tests/log/'