How to add a L2 regularization term in my loss function

autograd
17

How do you experiment with different values for weight_decay? So that you could show the amounts of regularization on the x axis and validation set performance on the y axis