torch.optim.lr_scheduler.get_last_lr
时间: 2023-12-12 11:03:47 浏览: 176
关于torch.optim的灵活使用详解(包括重写SGD,加上L1正则)
`torch.optim.lr_scheduler.get_last_lr()` 是 PyTorch 中的一个函数,它用于获取最近一次优化器的学习率。具体来说,它返回一个列表,其中包含每个分组的学习率,这些分组是通过 `torch.optim.Optimizer` 的 `param_groups` 参数指定的。如果您的优化器只有一个参数组,则该函数将返回一个包含一个学习率的列表。请注意,此函数仅在 PyTorch 版本大于等于 1.4.0 时才可用。您可以在 PyTorch 官方文档中找到更多关于此函数的信息。
阅读全文