from torch.utils.tensorboard import SummaryWriterif __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参数避免和其他人的端口一样卡住