L1正则化: 权重系数的绝对值之和被添加到成本中。通常称为权重的L1范数。
L2正则化 :所有权重系数的平方和被添加到成本中。通常称为权重的L2范数

Pytorch 提供了一种使用L2正则化的简单方法,就是通过在优化器中启用weight_decay 参数:

model = Architecturel(10,20,2)

opotimizer = torch.optim.Adam(model.parameters(),lr=1e-4,weight_decay=1e-5)

参考:Pytorch 深度学习

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐