torch.optim.lr_scheduler.steplr
时间: 2023-04-29 21:05:18 浏览: 98
pytorch 实现学习率设置 观察数据具体调整
torch.optim.lr_scheduler.StepLR 是 PyTorch 中用来调整学习率的一种方式。它会在给定的间隔周期内将学习率降低固定倍数。例如,如果我们将学习率设置为 0.1,降低倍数设置为 0.1,并将间隔周期设置为 2,那么在第 2 个周期、第 4 个周期和第 6 个周期时,学习率将被降低为 0.1 * 0.1 = 0.01。
阅读全文