Parameter dim in weight_norm with GRU

Hi,

I am wondering how to set the ‘dim’ parameter in ‘weight_norm’ when used with a GRU. My code:

rnn1 = weight_norm(weight_norm(nn.GRU(input_size=3, hidden_size=4), name=‘weight_hh_l0’), name=‘weight_ih_l0’),