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

Fix `cudaSetDevice` for CUDA 12 #49

Closed rusty1s closed 6 months ago