Error when running the link prediction task using c-rgcn

[Sidharth3301]

Using the default config file present in /configs/c-rgcn/lp-WN18.yaml after changing the weight initialisation as suggested in #12, I get the following error.

INFO - R-GCN Link Prediction  - Started
Start training...
ERROR - R-GCN Link Prediction  - Failed after 0:00:02!
Traceback (most recent calls WITHOUT Sacred internals):
  File "experiments/predict_links.py", line 151, in train
    predictions, penalty = model(graph, batch_idx)
  File "/usr/local/envs/torch_rgcn_venv/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in __call__
    result = self.forward(*input, **kwargs)
  File "/content/drive/MyDrive/research/dr/torch-rgcn/torch_rgcn/models.py", line 235, in forward
    x = self.rgc1(graph, features=x)
  File "/usr/local/envs/torch_rgcn_venv/lib/python3.7/site-packages/torch/nn/modules/module.py", line 541, in __call__
    result = self.forward(*input, **kwargs)
  File "/content/drive/MyDrive/research/dr/torch-rgcn/torch_rgcn/layers.py", line 550, in forward
    fw = torch.einsum('ni, rio -> rno', features, weights).contiguous()
  File "/usr/local/envs/torch_rgcn_venv/lib/python3.7/site-packages/torch/functional.py", line 201, in einsum
    return torch._C._VariableFunctions.einsum(equation, operands)
RuntimeError: size of dimension does not match previous size, operand 1, dim 1

Any help is appreciated, TIA!

[nicolas-hbt]

Hi, did you find any solution? I am encountering this very same issue.