I have a similar need to have different regularization factor for different layers.
The post you have referred here calculates the regularization loss using the L1 norm of each parameter. But they add all the regularization_loss to the final loss. And during back propagation, this loss is common to all the parameters and they are not different for different parameters.  in my case how to add different regularization factor for different layers?