TensorBoard 是 TensorFlow 提供的可视化工具,帮助开发者监控机器学习训练过程。以下是快速上手指南:
安装与启动
- 安装
pip install tensorboard
- 启动
在训练代码中添加以下行以生成日志:
然后运行:writer = tf.summary.create_file_writer('logs/your_experiment')
📌 访问 TensorBoard 官方文档 获取更多配置选项tensorboard --logdir=logs
基本用法
- 查看日志
打开浏览器访问http://localhost:6006
(默认端口) - 常用面板
- Scalars:监控损失函数/准确率等标量数据
- Graphs:可视化计算图
- Histograms:查看权重分布
- Images:显示训练过程中的图像
- Projector:三维数据可视化
常用功能示例
- 记录指标:
with writer.as_default(): tf.summary.scalar('loss', current_loss, step=epoch)
- 生成图像:
tf.summary.image('input', images, step=epoch)