optim.lr_scheduler.StepLR学习

torch.optim.lr_scheduler.StepLR 是 PyTorch 中的一个学习率调整器,按照一定的步长调整学习率。

torch.optim.lr_scheduler.StepLR(optimizer, step_size, gamma=0.1, last_epoch=-1, verbose=False)
  • optimizer:要使用该调度器的优化器。
  • step_size:每隔多少个 epoch 进行一次学习率调整。
  • gamma:学习率调整的乘数因子,即学习率乘以 gamma 得到新的学习率。
  • last_epoch:上一次的 epoch 数,如果不指定则默认为 -1。
  • verbose:是否打印学习率调整的信息,默认为 False。

在 PyTorch 中,学习率调度器一般与优化器一起使用,实现在训练过程中动态调整学习率。

1.构建模型: 构建神经网络模型。

2.定义优化器: 创建一个优化器(如 SGD 或 Adam)并将模型的参数传递给它,近年来Adam使用较多。

3.定义学习率调度器StepLR: 创建一个学习率调度器,并将其与优化器关联。

在训练循环中使用: 在每个训练迭代(或每个 epoch)结束时,调用学习率调度器的 step()方法。

 # 优化器
optimizer = optim.Adam([
        {'params': task_params},
        {'params': model.bert.parameters(), 'lr': config['bert_learning_rate']}],
        lr=config['task_learning_rate'],
        eps=config['adam_eps'])
#调度器
scheduler = optim.lr_scheduler.StepLR(optimizer,
                                          step_size=int(train_steps * 0.1))
#迭代训练
for _ in trange(int(config["num_epochs"]), desc="Epoch"):
    random.shuffle(examples)
    scheduler.step()

相关推荐

  1. 学习 学习

    2024-04-26 01:12:04       61 阅读
  2. 学期学习计划

    2024-04-26 01:12:04       41 阅读
  3. 学习笔记:机器学习

    2024-04-26 01:12:04       78 阅读
  4. C++学习-List学习

    2024-04-26 01:12:04       46 阅读
  5. opencv学习 机器学习

    2024-04-26 01:12:04       56 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-04-26 01:12:04       70 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-04-26 01:12:04       74 阅读
  3. 在Django里面运行非项目文件

    2024-04-26 01:12:04       62 阅读
  4. Python语言-面向对象

    2024-04-26 01:12:04       72 阅读

热门阅读

  1. 洛谷 P5960 [模板] 差分约束 题解 SPFA

    2024-04-26 01:12:04       31 阅读
  2. Thread 类的基本用法

    2024-04-26 01:12:04       24 阅读
  3. 流程图画图规范

    2024-04-26 01:12:04       26 阅读
  4. 浅谈Spring中的事务【Transactional】

    2024-04-26 01:12:04       28 阅读
  5. 【Python】使用Python连接ClickHouse进行批量数据写入

    2024-04-26 01:12:04       25 阅读
  6. nlp(6)--构建找规律模型任务

    2024-04-26 01:12:04       26 阅读
  7. 练习-字符串逆序统计

    2024-04-26 01:12:04       31 阅读
  8. 学习redis知识点

    2024-04-26 01:12:04       30 阅读
  9. Linux 静态IP地址修改与报错处理

    2024-04-26 01:12:04       28 阅读