首页 / 资讯中心 / 文章详情

PyTorch StepLR学习率调度:固定步长衰减实战解析

PyTorch StepLR学习率调度:固定步长衰减实战解析 ★ FEATURED ARTICLE
训练时明明已经把loss压下去了结果几十个epoch后突然在验证集上炸了或者训练集都快满分了、验证集却纹丝不动——这种场面我见过太多次。问题常常不在模型结构也不在数据预处理而是学习率从头到尾一成不变。今天这篇就专门聊聊torch.optim.lr_scheduler.StepLRPyTorch里最基础、也最容易上手的按固定步长衰减学习率的调度器弄清楚它的工作机制、参数含义和常见坑基本就够把训
阅读完成 · 觉得有帮助?
咨询建站