def _reset_parameters(self):
for weight in self.gru1.parameters():
my_init.rnn_init(weight)
for weight in self.gru2.parameters():
my_init.rnn_init(weight)
def _reset_parameters(self):
for weight in self.parameters():
my_init.default_init(weight)
if self.cover_size is not None:
for weight in self.gru.parameters():
my_init.rnn_init(weight)