def _reset_parameters_(self):
    self.entity_embeddings.reset_parameters()
    self.relation_embeddings.reset_parameters()
    for module in [self.linear1, self.linear2, self.bn1, self.bn2]:
        module.reset_parameters()
def _reset_parameters_(self):
    self.entity_embeddings.reset_parameters()
    self.relation_embeddings.reset_parameters()
    for module in [self.linear1, self.linear2, self.bn1, self.bn2]:
        module.reset_parameters()