I tried to run the example of "wn18_weighted", but had the following error:
File "....../dgl-ke-master/python/dglke/models/pytorch/loss.py", line 80, in get_total_loss
neg_loss = self._get_neg_loss(neg_score) * edge_weight
RuntimeError: The size of tensor a (128) must match the size of tensor b (2048) at non-singleton dimension 1
Should "neg_loss = self._get_neg_loss(neg_score) edge_weight" be changed to "neg_loss = ((self._get_neg_loss(neg_score).T) edge_weight).T"?
Hi,
I tried to run the example of "wn18_weighted", but had the following error:
File "....../dgl-ke-master/python/dglke/models/pytorch/loss.py", line 80, in get_total_loss neg_loss = self._get_neg_loss(neg_score) * edge_weight RuntimeError: The size of tensor a (128) must match the size of tensor b (2048) at non-singleton dimension 1
Should "neg_loss = self._get_neg_loss(neg_score) edge_weight" be changed to "neg_loss = ((self._get_neg_loss(neg_score).T) edge_weight).T"?