pytorch使用tensorboard绘制曲线如何设置曲线颜色

在 PyTorch 中，可以使用 `torch.utils.tensorboard` 模块来绘制曲线。要设置曲线颜色，可以通过在 `add_scalar` 函数中指定 `tag` 和 `global_step` 参数，以及 `value` 参数来记录某个指标的值。同时，可以使用 `add_custom_scalars_multi` 函数来自定义曲线的颜色和标签。以下是一个示例代码：

from torch.utils.tensorboard import SummaryWriter
import torch

# 创建 SummaryWriter 对象
writer = SummaryWriter('log_dir')

# 记录训练集和测试集上的 loss 值
for i in range(100):
    train_loss = torch.rand(1) * 0.2 + 0.8
    val_loss = torch.rand(1) * 0.2 + 1.0
    writer.add_scalar('train/loss', train_loss, i)
    writer.add_scalar('val/loss', val_loss, i)

# 自定义曲线的颜色和标签
colors = {'train': 'red', 'val': 'blue'}
labels = {'loss': ['train_loss', 'val_loss']}
writer.add_custom_scalars_multi(tags=labels, scalars=colors)

# 关闭 SummaryWriter 对象
writer.close()

在运行 TensorBoard 后，可以在 Scalars 标签下看到两条曲线，通过点击右侧的颜色方块，可以修改曲线的颜色。