loss = mse(pred, target) l1 = 0 for p in net.parameters(): l1 = l1 + p.abs().sum() loss = loss + lambda_l1 * l1 loss.backward() optimizer.step()