pytorch_spline_conv:在PyTorch中实现基于样条的SplineCNN卷积运算符 源码
基于样条的SplineCNN卷积算子 这是SplineCNN基于样条的卷积运算符的PyTorch实现,如我们的论文所述: Matthias Fey,Jan Eric Lenssen,Frank Weichert,HeinrichMüller: (CVPR 2018) 该运算符适用于所有浮点数据类型,并且已针对CPU和GPU实施。 安装 二进制文件 我们提供适用于所有主要OS / PyTorch / CUDA组合的拨轮,请参见。 PyTorch 1.7.0 要为PyTorch 1.7.0安装二进制文件,只需运行 pip install torch-spline-conv -f https
用户评论