Hi everyone,
I’m trying to apply newly added weight_norm to nn.GRU.
But it seems weight_norm only accept single name (default: ‘weight’) per call.
As you know, nn.GRU has #(layer x 2) separate weight variables named weight_ih_l[k] and weight_hh_l[k].
I wonder what would be most elegant way to apply the weight_norm to all those weight variables.
Thanks!