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. 在笔记本单元格中添加魔法命令 %tensorboard ，以启用 TensorBoard：

   %load_ext tensorboard %tensorboard --logdir=runs