How to add a L2 regularization term in my loss function

autograd
2

set “weight_decay” parameter to a non zero value in your optimizer(sgd, adam, …)(it’s the alpha in your equation)
edit: I think it’s alpha times two actually

4 Likes