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

时间:2024-03-07 16:41:07
【文件属性】:

文件名称:pytorch_spline_conv:在PyTorch中实现基于样条的SplineCNN卷积运算符

文件大小:31KB

文件格式:ZIP

更新时间:2024-03-07 16:41:07

pytorch geometric-deep-learning spline-cnn graph-neural-networks C++

基于样条的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://pytorch-geometric.com/whl/torch-1.7.0+${CUDA}.html 其中${CUDA}应该由cpu , cu92 , cu101 , cu102或cu110 c


【文件预览】:
pytorch_spline_conv-master
----MANIFEST.in(85B)
----.travis.yml(6KB)
----.coveragerc(119B)
----cmake()
--------TorchSplineConvConfig.cmake.in(1KB)
----torch_spline_conv()
--------weighting.py(298B)
--------__init__.py(1KB)
--------conv.py(3KB)
--------basis.py(389B)
----test()
--------test_basis.py(1KB)
--------utils.py(293B)
--------__init__.py(0B)
--------test_conv.py(3KB)
--------test_weighting.py(2KB)
----LICENSE(1KB)
----setup.cfg(107B)
----setup.py(3KB)
----.gitignore(68B)
----CMakeLists.txt(2KB)
----README.md(7KB)
----script()
--------cuda.sh(5KB)
--------torch.sh(727B)
--------gcc.sh(388B)
--------conda.sh(703B)
--------rename_wheel.py(655B)
----csrc()
--------cuda()
--------spline_conv.h(384B)
--------weighting.cpp(4KB)
--------version.cpp(369B)
--------basis.cpp(3KB)
--------cpu()

网友评论