TensorBoard 支持

在开始之前

• 确保已安装 Tensorboard 包。

启用 TensorBoard

1. 使用 TensorFlow 或 PyTorch 保存您的训练指标或日志。 例如：

   from torch.utils.tensorboard import SummaryWriter writer = SummaryWriter('runs/experiment_1') for i in range(10): writer.add_scalar('Loss/train', i * 0.1, i) writer.close()

   此示例会在 runs 目录中生成日志文件。

2. 通过在 notebook 单元格中添加魔法命令 %tensorboard 启用 TensorBoard：

   %load_ext tensorboard %tensorboard --logdir=runs