benedekrozemberczki / ClusterGCN

A PyTorch implementation of "Cluster-GCN: An Efficient Algorithm for Training Deep and Large Graph Convolutional Networks" (KDD 2019).
GNU General Public License v3.0
786 stars 135 forks source link

ImportError: No module named 'torch_spline_conv' #12

Closed 1byxero closed 4 years ago

1byxero commented 4 years ago

I followed the instructions of installation properly, however, error above occurred.

After checking the site packages folder, i do not find the file torch_spline_conv. I will google around for finding out why that is happening, but thought you might have some insights

Any help is appreciated.

The complete trace is as follows

File "src/main.py", line 4, in <module>
    from clustergcn import ClusterGCNTrainer
  File "/media/anuj/Softwares & Study Material/Study Material/MS Stuff/RA/ClusterGCN/src/clustergcn.py", line 5, in <module>
    from layers import StackedGCN
  File "/media/anuj/Softwares & Study Material/Study Material/MS Stuff/RA/ClusterGCN/src/layers.py", line 2, in <module>
    from torch_geometric.nn import GCNConv
  File "/home/anuj/virtualenv-forest/gcn/lib/python3.5/site-packages/torch_geometric/nn/__init__.py", line 1, in <module>
    from .conv import *  # noqa
  File "/home/anuj/virtualenv-forest/gcn/lib/python3.5/site-packages/torch_geometric/nn/conv/__init__.py", line 1, in <module>
    from .spline_conv import SplineConv
  File "/home/anuj/virtualenv-forest/gcn/lib/python3.5/site-packages/torch_geometric/nn/conv/spline_conv.py", line 3, in <module>
    from torch_spline_conv import SplineConv as Conv
ImportError: No module named 'torch_spline_conv'
1byxero commented 4 years ago

After a some googling around and pulling my hair, I have found out that pytorch geometric has made torch_spline_conv as optional requirement. After installing it, the training stage started!

benedekrozemberczki commented 4 years ago

Yeah, sometime there are major changes that they do. Also the METIS wrapper does not work with NetworkX 2.0.