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)
def init_weights(self):
for layer in self.dense:
weight_init(layer)
def init_weights(self):
for key, layer in self.embed.items():
weight_init(layer)