Exemplo n.º 1
0
 def init_weights(self):
     for key, layer in self.sparse_weight.items():
         weight_init(layer)
     for key, layer in self.dense_weight.items():
         weight_init(layer)
Exemplo n.º 2
0
 def init_weights(self):
     for layer in self.dense:
         weight_init(layer)
Exemplo n.º 3
0
 def init_weights(self):
     for key, layer in self.embed.items():
         weight_init(layer)