Hi, I do not see any requirement to install torch geometric. So when I ran your demo.ipynb, it caused an error
'No module named 'torch_geometric.nn.dense.linear'
What is the version of torch geometric you used for this repo? I'm using torch geometric 1.7.0
Thanks