22.自定义变量的Tensorboard可视化

代码和上一章差不多,我们在上一章的基础上进行更改

目录

1  随epoch更改学习速率

2  将变化情况写在文件中


1  随epoch更改学习速率

我们当前想让学习速率随epoch的变化而变化,首先我们定义一个learning_rate随时间变化的函数

然后我们创建一个回调函数,使用tf.keras.callbacks.LearningRateScheduler()参数是上面创建的函数

然后把回调函数放在callbacks的列表中

此时进行训练我们还不能在tensorboard中显示learning_rate,我们需要将变化情况写入文件中

2  将变化情况写在文件中

首先创建一个写对象,参数为要将文件存放的路径

然后将其设置为默认写对象

然后我们再定义的变化learning_rate函数上加上tf.summary.scalar()以记录变量

第一个learning_rate是要记录的名称,第二个learning_rate是要记录的内容(我们上面的变量),第三个是横坐标epoch

为了体现学习速率的变化,我们训练7个epoch

训练后我们发现文件夹中出现了lr

打开tensorboard,我们发现出现了learning_rate这一项,但是我们可以发现learning_rate的变化的渐变的而不是直接改变的,所以我们如果要更好的控制learning_rate,我们需要将epoch的间隔拉大一点,让他可以顺利渐变

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
使用PyTorch可视化TensorBoard的步骤如下: 1. 安装TensorBoard ``` pip install tensorboard ``` 2. 在PyTorch中创建SummaryWriter对象 ```python from torch.utils.tensorboard import SummaryWriter # 创建SummaryWriter对象 writer = SummaryWriter('runs/experiment_name') ``` 其中,`experiment_name`为实验名称,可以自定义。这里通过`runs`目录来存储实验数据。 3. 在训练或验证的过程中添加可视化代码 在代码中使用`writer.add_scalar()`方法来添加可视化代码,方法的参数包括要可视化变量名、变量值和步数。例如: ```python # 添加训练损失 writer.add_scalar('Train/Loss', loss.item(), step) # 添加验证准确率 writer.add_scalar('Validation/Accuracy', accuracy, step) ``` 4. 启动TensorBoard 在命令行中使用以下命令启动TensorBoard: ``` tensorboard --logdir runs ``` 其中,`--logdir`参数指定实验数据所在的目录。在上面的代码中,我们将实验数据保存在`runs`目录下。 5. 在浏览器中查看可视化结果 在浏览器中输入以下地址查看TensorBoard可视化结果: ``` http://localhost:6006/ ``` 这里是一个可视化变量曲线的代码示例: ```python import torch from torch.utils.tensorboard import SummaryWriter # 创建SummaryWriter对象 writer = SummaryWriter('runs/experiment_name') # 创建模拟数据 x = torch.linspace(-5, 5, 100) y = torch.sin(x) # 可视化曲线 for i in range(100): writer.add_scalar('Sin(x)', y[i], i) # 关闭SummaryWriter对象 writer.close() ``` 这段代码将生成一条正弦曲线,并将其可视化TensorBoard中。在TensorBoard中,我们可以选择`Scalars`选项卡来查看变量曲线。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Suyuoa

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值