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參數避免和其他人的端口一樣卡住