Pass arguments from the weight decay functions down to the layer.