TensorBoard 是 TensorFlow 提供的可视化工具,帮助开发者监控机器学习训练过程。以下是快速上手指南:

安装与启动

  1. 安装
    pip install tensorboard
    
  2. 启动
    在训练代码中添加以下行以生成日志:
    writer = tf.summary.create_file_writer('logs/your_experiment')
    
    然后运行:
    tensorboard --logdir=logs
    
    📌 访问 TensorBoard 官方文档 获取更多配置选项

基本用法

  • 查看日志
    打开浏览器访问 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)
    
TensorBoard_Overview

扩展阅读

TensorBoard_Usage_Steps