pytorch中的多种scheduler
pytorch中的多种schedulerLambdaLR公式lrepoch=lrinitial∗Lambda(epoch)lr_{epoch}=lr_{initial}∗Lambda(epoch)lrepoch=lrinitial∗Lambda(epoch)代码import torchimport matplotlib.pyplot as pltmodel = torch.nn.Linear(2, 1)optimizer = torch.optim.SGD(model.
原创
2021-12-06 17:10:27 ·
1535 阅读 ·
0 评论