文件名称：spconv:PyTorch中的空间稀疏卷积
文件大小：256KB
文件格式：ZIP
更新时间：2024-05-20 20:54:01
C++
SpConv：PyTorch空间稀疏卷积库 这是一个空间稀疏的卷积库，例如但速度更快且易于阅读。 该库提供稀疏卷积/转置，子流形卷积，逆卷积和稀疏maxpool。 2020-5-2，我们在此版本的spconv中添加了ConcatTable，JoinTable，AddTable和Identity函数，以构建ResNet和Unet。 码头工人： docker pull scrin/dev-spconv ，包含python 3.8，cuda 10.1，鱼壳，最新的pytorch和tensorflow。 在Ubuntu 16.04 / 18.04上安装 如果您使用的是pytorch 1.4+，并且遇到“ nvcc致命：未知-Wall”，则需要转到割炬软件包目录并在Caffe2Targets.cmake的INTERFACE_COMPILE_OPTIONS中删除包含“ -Wall”的标志。 此项
【文件预览】：
spconv-master
----.github()
--------workflows()
----.gitmodules(309B)
----spconv()
--------identity.py(361B)
--------utils()
--------pool.py(3KB)
--------ops.py(6KB)
--------tables.py(1KB)
--------__init__.py(4KB)
--------modules.py(7KB)
--------functional.py(5KB)
--------test_utils.py(8KB)
--------conv.py(19KB)
----src()
--------utils()
--------spconv()
--------cuhash()
----Dockerfile(648B)
----PERFORMANCE_GUIDE.md(2KB)
----CMakeLists.txt(2KB)
----example()
--------mnist_sparse.py(6KB)
----LICENSE(11KB)
----test()
--------fake_train.py(5KB)
--------src()
--------CMakeLists.txt(1KB)
--------test_conv.py(30KB)
--------fake_dist_train.py(5KB)
----setup.py(5KB)
----README.md(6KB)
----format_all.sh(271B)
----.gitignore(1KB)
----third_party()
--------catch2()
--------mp11()
--------cutlass()
--------pybind11()
----CHANGELOG.md(693B)
----include()
--------tensorview()
--------tensorrt()
--------spconv()
--------cuhash()
--------utility()
--------torch_utils.h(4KB)
--------paramsgrid.h(2KB)
--------tsl()