tensorboard summary使用

本文介绍如何在TensorFlow中使用summary模块记录训练过程中的关键指标,如loss,并将其可视化。通过定义scalar变量和使用File_Writer,可以将训练过程中每个epoch的loss变化保存下来,便于后续分析和优化。
部署运行你感兴趣的模型镜像

假设想要输出loss随epoch的变化图:

1、构建图过程中声明scalar变量

在图中定义loss节点后,加一个

loss_summary = tf.summry.scalar('loss',loss)


当有多个要输出时,可以用

merged_summary_op = tf.merge_all_summaries()

汇总到一个节点,执行时执行这个节点,add这个节点就行

2、构建完整个图后加File_Writer

构建完图,在执行前,声明

file_writer = tf.summary.FileWriter(logdir,tf.get_default_graph())

比如在全局变量init语句之后

3 、执行过程中

喂入数据后run上面的loss_summary节点
然后:

file_writer.add_summary(loss_summary_str,epoch)

4、查看

您可能感兴趣的与本文相关的镜像

GPT-oss:20b

GPT-oss:20b

图文对话
Gpt-oss

GPT OSS 是OpenAI 推出的重量级开放模型,面向强推理、智能体任务以及多样化开发场景

### TensorBoard 使用教程与模型可视化指南 TensorBoard 是 TensorFlow 提供的可视化工具,广泛用于深度学习模型的训练监控和结果展示。它支持多种数据类型的可视化,包括标量、图像、直方图、文本、音频等,并且可以通过插件扩展功能。以下是一个基础的使用教程和模型可视化指南。 #### 安装 TensorBoard使用 TensorBoard 之前,需要确保已经安装了 TensorFlow 或者 PyTorch(配合 TensorBoardX 使用)。对于 TensorFlow 用户,TensorBoard 已经内置,可以直接使用: ```bash pip install tensorflow ``` 对于 PyTorch 用户,可以使用 TensorBoardX 或 PyTorch 的内置 TensorBoard 支持: ```bash pip install tensorboardX # 或者使用 PyTorch 1.2+ 的内置支持 pip install torch ``` #### 基本使用流程 1. **导入 TensorBoard 模块** 在代码中导入 TensorBoardSummaryWriter 模块: ```python from torch.utils.tensorboard import SummaryWriter # 或者对于 TensorFlow 用户 from tensorflow.summary import create_file_writer ``` 2. **创建日志文件夹** 创建一个文件夹来存储 TensorBoard 的日志文件: ```python writer = SummaryWriter('runs/experiment_1') ``` 3. **记录数据** 在训练过程中,可以记录标量、图像、直方图等数据: ```python # 记录标量 writer.add_scalar('Loss/train', loss.item(), epoch) # 记录图像 writer.add_image('Image/sample', image_tensor, epoch) # 记录直方图 writer.add_histogram('Weights/conv1', weights, epoch) ``` 4. **启动 TensorBoard** 在终端中运行以下命令启动 TensorBoard: ```bash tensorboard --logdir=runs ``` 打开浏览器,访问 `http://localhost:6006`,即可看到可视化的训练过程。 #### 高级功能 - **图像可视化**:可以展示模型处理的图像数据,帮助分析模型的输入输出[^3]。 - **嵌入投影**:用于可视化高维向量(如词嵌入),支持 t-SNE 和 PCA 降维方法[^5]。 - **注意力热图**:在 NLP 任务中,可以展示模型的注意力分布,验证模型是否理解语言结构[^5]。 #### TensorBoardX 的使用 对于 PyTorch 用户,TensorBoardX 是一个常用的扩展工具,提供了与 TensorBoard 兼容的 API: ```python from tensorboardX import SummaryWriter writer = SummaryWriter('runs/experiment_1') writer.add_scalar('Loss/train', loss.item(), epoch) writer.close() ``` #### 注意事项 - **多 GPU 支持**:TensorBoard 在多 GPU 环境下可能存在限制,TensorBoardX 提供了更好的支持[^2]。 - **性能优化**:在处理大规模数据时,建议合理设置日志记录频率,避免性能下降[^2]。 --- ###
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值