weight_norm is applied on modules not parameters, so it won’t work in your code snippet.
Internally, it will register new parameters as seen here 1 so you could try to adapt this code to your use case and manipulate your parameter manually.