You could adapt the for loop and add the desired regularization to the corresponding layer.
In the current example, the code just iterates all parameters, so you could add some if conditions etc. (depends on your use case).