pytorch tensorboard基本用法整理

from torch.utils.tensorboard import SummaryWriter

if __name__ == '__main__':
    aa = SummaryWriter('logs') # 创建保存了summarywriter的log目录
    for i in range(100):
        aa.add_scalar('y = x', i, i)  # 后两个参数先y轴后x轴 x轴往往是global step y轴用于输出loss或者其他需要观察的变量值

要查看保存的tensorboard文件需要在命令行中输入：

tensorboard --logdir=PATH --port 6007 

其中PATH是保存了events的目录的绝对路径或是相对路径
通过设置port参数避免和其他人的端口一样卡住