在Pytorch中应用权重正则化
L1正则化: 权重系数的绝对值之和被添加到成本中。通常称为权重的L1范数。L2正则化 :所有权重系数的平方和被添加到成本中。通常称为权重的L2范数Pytorch 提供了一种使用L2正则化的简单方法,就是通过在优化器中启用weight_decay 参数:model = Architecturel(10,20,2)opotimizer = torch.optim.Adam(model.paramete
·
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 深度学习
更多推荐
已为社区贡献2条内容
所有评论(0)