1. 首页
  2. 数据库
  3. 其它
  4. pytorch_spline_conv:在PyTorch中实现基于样条的SplineCNN卷积运算符 源码

pytorch_spline_conv:在PyTorch中实现基于样条的SplineCNN卷积运算符 源码

上传者: 2021-02-22 08:36:05上传 ZIP文件 30.7KB 热度 150次
基于样条的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
用户评论