deepakanandece / pytorch_geometric Goto Github PK

PyTorch Geometric is a Geometric Deep Learning extension library for PyTorch. In addition, PyTorch Geometric contains the spline-based convolutional layer from our paper SplineCNN: Fast Geometric Deep Learning with Continuous B-Spline Kernels.

Documentation

Install pytorch and torchvision manually (Installation). If cuda is available it is recommended to install cupy (Installation).

To install the additional python packages needed, run:

pip install -r requirements.txt

Install the test requirements and run the test suite:

pip install -r requirements_test.txt
nosetests

Please cite our paper if you use this code in your own work:

@article{Fey2017,
  title={{SplineCNN}: Fast Geometric Deep Learning with Continuous B-Spline Kernels},
  author={Matthias Fey and {Jan Eric} Lenssen and Frank Weichert and Heinrich M{\"u}ller},
  journal={CoRR},
  volume={abs/1711.08920},
  year={2017},
}