pytorch中使用tensorboard绘制Accuracy/Loss曲线（train和test显示在同一幅图中）

因为tensorboard可以在同一幅图中显示不同文件夹下的曲线，所以将train和test分别存到不同的文件夹里就可以在同一副图中展示，简要记录代码。

from torch.utils.tensorboard import SummaryWriter

# 将 train accuracy 保存到 "tensorboard/train" 文件夹
log_dir = os.path.join('tensorboard', 'train')
train_writer = SummaryWriter(log_dir=log_dir)
# 将 test accuracy 保存到 "tensorboard/test" 文件夹
log_dir = os.path.join('tensorboard', 'test')
test_writer = SummaryWriter(log_dir=log_dir)

# 绘制
train_writer.add_scalar('Accuracy', train_accuracy, epoch)
test_writer.add_scalar('Accuracy', test_accuracy, epoch)

输入tensorboard --logdir=tensorboard，就可以得到类似下图的结果