I am just using the standard pytorch weight_norm functionality (https://pytorch.org/docs/master/_modules/torch/nn/utils/weight_norm.html 8). It takes module as input so not sure m.weight.data will work. But I do like the is instance(…) better than my re.findall hack.
Thanks for  that.