Esempio n. 1
0
 def reset_parameters(self):
     reset(self.nn)
     uniform(self.in_channels, self.root)
     uniform(self.in_channels, self.bias)
Esempio n. 2
0
 def reset_parameters(self):
     reset(self.gate_nn)
     reset(self.nn)
Esempio n. 3
0
 def reset_parameters(self):
     reset(self.encoder)
     reset(self.decoder)