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)