This repository has been archived by the owner on Dec 19, 2023. It is now read-only.

why only update tail entities?Is there an error here #41

Open

jweihe opened this issue Apr 10, 2022 · 0 comments

jweihe commented Apr 10, 2022

` mask_indices = torch.unique(batch_inputs[:, 2]).cuda()
mask = torch.zeros(self.entity_embeddings.shape[0]).cuda()
mask[mask_indices] = 1.0

    entities_upgraded = self.entity_embeddings.mm(self.W_entities)
    out_entity_1 = entities_upgraded + \
        mask.unsqueeze(-1).expand_as(out_entity_1) * out_entity_1

`

The text was updated successfully, but these errors were encountered:

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.