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.
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},
}