spconv:PyTorch中的空间稀疏卷积

时间:2021-05-01 03:07:21
【文件属性】:
文件名称:spconv:PyTorch中的空间稀疏卷积
文件大小:256KB
文件格式:ZIP
更新时间:2021-05-01 03:07:21
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()

网友评论