pytorch调用tensorboard方法尝试

本文介绍如何在PyTorch中使用TensorBoard监视训练损失,包括安装必要软件包、配置SummaryWriter、添加监视变量及启动TensorBoard的方法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

tensorboard提供了用于监视训练损失很好的接口,可以帮助我们更好的调整参数。下文介绍如何在pytorch中调用tensorboard。

首先,安装tensorboard、tensorflow以及tensorboardX

第二,在文件开头导入SummaryWriter

from tensorboardX import SummaryWriter

第三,同tensorflow的tensorboard一样,tensorboardX提供多种记录方式如scalar、image等。

writer = SummaryWriter('path')

如果不添加path,则默认以时间命名。

第四,添加监视变量

writer.add_scalar('Train/Acc', Acc, iter)

第五,打开tensorboard

tensorboard --logdir 'path'

第六,在浏览器打开6006端口

 

 

 

 

 

### 安装 TensorBoard 在 macOS 上安装 TensorBoard 可通过 `pip` 工具完成。以下是具体方法: 确保已安装 Python 和 pip,然后运行以下命令来安装最新版本的 TensorBoard: ```bash pip install --upgrade tensorboard ``` 如果正在使用虚拟环境,请激活该环境后再执行上述命令[^1]。 对于更高级的功能需求,可以考虑安装第三方库 TensorBoardX 来支持 PyTorch 或其他框架的日志记录功能: ```bash git clone https://github.com/lanpa/tensorboardX && cd tensorboardX && python setup.py install ``` ### 启动 TensorBoard 假设已经有一个日志目录(例如 `/Users/your_name/logs`),可以通过如下方式启动 TensorBoard: #### 步骤说明 创建 TensorFlow 图并将其写入指定的日志目录中: ```python import tensorflow as tf logdir = "/path/to/log_directory" tf.compat.v1.reset_default_graph() with tf.Graph().as_default(): input1 = tf.constant([1.0, 2.0, 3.0], name="input1") input2 = tf.Variable(tf.random.uniform([3]), name="input2") output = tf.add_n([input1, input2], name="add") with tf.compat.v1.Session() as sess: writer = tf.compat.v1.summary.FileWriter(logdir, sess.graph) writer.close() ``` 随后,在终端中输入以下命令以启动 TensorBoard 并指向日志目录: ```bash tensorboard --logdir=/path/to/log_directory ``` 默认情况下,TensorBoard 将监听端口 6006,并可通过浏览器访问地址 `http://localhost:6006` 查看可视化界面[^3]。 需要注意的是,当遇到无法找到正确 TensorBoard 命令的情况时,可能是由于系统优先调用了全局路径中的二进制文件而非当前虚拟环境下的工具。此时可尝试显式指定完整路径或重新配置 PATH 环境变量[^2]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值