GNN Explainer example not running

[ZhaoShuQing0102]

hello dear,

I tried to run gnn_explainer example. I am getting the following error.

Traceback (most recent call last):
File "E:/GNNCode/MyGNN/main.py", line 48, in
edge_weight=edge_weight)
File "D:\conda\envs\GNNExplainer\lib\site-packages\torch_geometric\nn\models\gnn_explainer.py", line 208, in explain_node
prediction = self.get_initial_prediction(x, edge_index, **kwargs)
File "D:\conda\envs\GNNExplainer\lib\site-packages\torch\autograd\grad_mode.py", line 28, in decorate_context
return func(*args, **kwargs)
File "D:\conda\envs\GNNExplainer\lib\site-packages\torch_geometric\nn\models\explainer.py", line 235, in get_initial_prediction
out = self.model(x, edge_index, **kwargs)
File "D:\conda\envs\GNNExplainer\lib\site-packages\torch\nn\modules\module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "E:/GNNCode/MyGNN/main.py", line 25, in forward
x = F.relu(self.conv1(x, edge_index, edge_weight))
File "D:\conda\envs\GNNExplainer\lib\site-packages\torch\nn\modules\module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "D:\conda\envs\GNNExplainer\lib\site-packages\torch_geometric\nn\conv\gcn_conv.py", line 195, in forward
size=None)
File "D:\conda\envs\GNNExplainer\lib\site-packages\torch_geometric\nn\conv\message_passing.py", line 317, in propagate
out = self.message(**msg_kwargs)
File "D:\conda\envs\GNNExplainer\lib\site-packages\torch_geometric\nn\conv\gcn_conv.py", line 203, in message
return x_j if edge_weight is None else edge_weight.view(-1, 1) * x_j
RuntimeError: The size of tensor a (13264) must match the size of tensor b (19) at non-singleton dimension 0

the size of edge_weight is 13264, the size of x_j is (19,16).

So what can I do to successfully run the GNN Explainer example.

Thank you.

[EdisonLeeeee]

This bug was resolved in PyG>2.0.4, you need to upgrade your PyG to a higher version.

[ZhaoShuQing0102]

Thanks