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

Cuda no cuda #3

Closed AntoinePrv closed 5 years ago

AntoinePrv commented 5 years ago

Not 100% sure about the warning I fixed. I marked the docstring as raw to avoid the undefined character \s in this line.