固定步长衰减策略
参数
step_szie:表示每隔多少个步长修改一次(训练的时候表示每隔多少个epoch修改一次)
gamma:0.1 表示每一次修改采用 lr_new = lr_old * gamma
optimizer = torch.optim.SGD(model.parameters(), lr=0.01)
scheduler_StepLR = torch.optim.lr_scheduler.StepLR(optimizer, step_size=10, gamma=0.9)
多步长衰减策略
参数
milestones 表示在哪个步长时衰减(epoch)
gamma 同上
optimizer = torch