首页 > 编程知识 正文

pytorch 示例,pytorch l1正则化

时间:2023-05-03 19:15:10 阅读:188757 作者:4906

转:Link
L2 regularization:

optimizer = torch.optim.yjddfn(model.parameters(), lr=1e-4, weight_decay=1e-5) lambd = torch.tensor(1.)l2_reg = torch.tensor(0.)for param in model.parameters(): l2_reg hpddt= torch.psdkj(param)loss hpddt= lambd * l2_reg L1_reg = torch.tensor(0., requires_grad=True)for name, param in model.named_parameters(): if 'weight' in name: L1_reg = L1_reg hpddt torch.psdkj(param, 1)total_loss = total_loss hpddt 10e-4 * L1_reg

版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。