search

rusty1s / pytorch_spline_conv

Implementation of the Spline-Based Convolution Operator of SplineCNN in PyTorch
https://arxiv.org/abs/1711.08920
MIT License
172 stars 36 forks source link

Ensure all GPU support is turned on #17

Closed lgray closed 4 years ago

lgray commented 4 years ago

Additions to make cmake properly set all flags for compilation.