spconv:PyTorch中的空间稀疏卷积

上传者: 42099302 | 上传时间: 2022-08-10 17:52:13 | 文件大小: 256KB | 文件类型: ZIP
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”的标志。 此项

文件下载

资源详情

[{"title":"( 96 个子文件 256KB ) spconv:PyTorch中的空间稀疏卷积","children":[{"title":"spconv-master","children":[{"title":".github","children":[{"title":"workflows","children":[{"title":"build.yaml <span style='color:#111;'> 1.05KB </span>","children":null,"spread":false}],"spread":true}],"spread":true},{"title":".gitmodules <span style='color:#111;'> 309B </span>","children":null,"spread":false},{"title":"spconv","children":[{"title":"identity.py <span style='color:#111;'> 361B </span>","children":null,"spread":false},{"title":"utils","children":[{"title":"__init__.py <span style='color:#111;'> 12.37KB </span>","children":null,"spread":false}],"spread":true},{"title":"pool.py <span style='color:#111;'> 3.33KB </span>","children":null,"spread":false},{"title":"ops.py <span style='color:#111;'> 6.48KB </span>","children":null,"spread":false},{"title":"tables.py <span style='color:#111;'> 1.39KB </span>","children":null,"spread":false},{"title":"__init__.py <span style='color:#111;'> 4.33KB </span>","children":null,"spread":false},{"title":"modules.py <span style='color:#111;'> 6.64KB </span>","children":null,"spread":false},{"title":"functional.py <span style='color:#111;'> 5.36KB </span>","children":null,"spread":false},{"title":"test_utils.py <span style='color:#111;'> 7.73KB </span>","children":null,"spread":false},{"title":"conv.py <span style='color:#111;'> 18.62KB </span>","children":null,"spread":false}],"spread":true},{"title":"src","children":[{"title":"utils","children":[{"title":"CMakeLists.txt <span style='color:#111;'> 1.35KB </span>","children":null,"spread":false},{"title":"all.cc <span style='color:#111;'> 5.56KB </span>","children":null,"spread":false},{"title":"nms.cu <span style='color:#111;'> 5.58KB </span>","children":null,"spread":false}],"spread":true},{"title":"spconv","children":[{"title":"indice.cu <span style='color:#111;'> 16.41KB </span>","children":null,"spread":false},{"title":"nms.cc <span style='color:#111;'> 4.90KB </span>","children":null,"spread":false},{"title":"reordering.cu <span style='color:#111;'> 15.40KB </span>","children":null,"spread":false},{"title":"indice.cc <span style='color:#111;'> 12.68KB </span>","children":null,"spread":false},{"title":"cublas_gemm.cc <span style='color:#111;'> 2.19KB </span>","children":null,"spread":false},{"title":"CMakeLists.txt <span style='color:#111;'> 803B </span>","children":null,"spread":false},{"title":"all.cc <span style='color:#111;'> 1.51KB </span>","children":null,"spread":false},{"title":"nms.cu <span style='color:#111;'> 2.51KB </span>","children":null,"spread":false},{"title":"pool_ops.cc <span style='color:#111;'> 2.52KB </span>","children":null,"spread":false},{"title":"maxpool.cu <span style='color:#111;'> 18.42KB </span>","children":null,"spread":false},{"title":"spconv_ops.cc <span style='color:#111;'> 23.70KB </span>","children":null,"spread":false},{"title":"maxpool.cc <span style='color:#111;'> 3.15KB </span>","children":null,"spread":false},{"title":"reordering.cc <span style='color:#111;'> 2.79KB </span>","children":null,"spread":false},{"title":"pillar_scatter.cu <span style='color:#111;'> 2.34KB </span>","children":null,"spread":false}],"spread":false},{"title":"cuhash","children":[{"title":"main.cc <span style='color:#111;'> 1.38KB </span>","children":null,"spread":false},{"title":"hash_table.cu <span style='color:#111;'> 4.52KB </span>","children":null,"spread":false},{"title":"debugging.cpp <span style='color:#111;'> 3.40KB </span>","children":null,"spread":false},{"title":"hash_functions.cpp <span style='color:#111;'> 351B </span>","children":null,"spread":false},{"title":"hash_table.cpp <span style='color:#111;'> 7.26KB </span>","children":null,"spread":false},{"title":"debugging.cu <span style='color:#111;'> 7.77KB </span>","children":null,"spread":false},{"title":"CMakeLists.txt <span style='color:#111;'> 1.14KB </span>","children":null,"spread":false},{"title":"hash_functions.cu <span style='color:#111;'> 1.15KB </span>","children":null,"spread":false}],"spread":true}],"spread":true},{"title":"Dockerfile <span style='color:#111;'> 648B </span>","children":null,"spread":false},{"title":"PERFORMANCE_GUIDE.md <span style='color:#111;'> 1.80KB </span>","children":null,"spread":false},{"title":"CMakeLists.txt <span style='color:#111;'> 2.17KB </span>","children":null,"spread":false},{"title":"example","children":[{"title":"mnist_sparse.py <span style='color:#111;'> 5.52KB </span>","children":null,"spread":false}],"spread":true},{"title":"LICENSE <span style='color:#111;'> 11.07KB </span>","children":null,"spread":false},{"title":"test","children":[{"title":"fake_train.py <span style='color:#111;'> 4.91KB </span>","children":null,"spread":false},{"title":"src","children":[{"title":"catch_main.cpp <span style='color:#111;'> 481B </span>","children":null,"spread":false},{"title":"test_conv_rule.cpp <span style='color:#111;'> 5.05KB </span>","children":null,"spread":false}],"spread":true},{"title":"CMakeLists.txt <span style='color:#111;'> 1.13KB </span>","children":null,"spread":false},{"title":"test_conv.py <span style='color:#111;'> 30.47KB </span>","children":null,"spread":false},{"title":"fake_dist_train.py <span style='color:#111;'> 5.33KB </span>","children":null,"spread":false}],"spread":true},{"title":"setup.py <span style='color:#111;'> 4.79KB </span>","children":null,"spread":false},{"title":"README.md <span style='color:#111;'> 6.38KB </span>","children":null,"spread":false},{"title":"format_all.sh <span style='color:#111;'> 271B </span>","children":null,"spread":false},{"title":".gitignore <span style='color:#111;'> 1.20KB </span>","children":null,"spread":false},{"title":"third_party","children":[{"title":"catch2","children":[{"title":"catch.hpp <span style='color:#111;'> 466.08KB </span>","children":null,"spread":false}],"spread":false},{"title":"mp11","children":null,"spread":false},{"title":"cutlass","children":null,"spread":false},{"title":"pybind11","children":null,"spread":false}],"spread":false},{"title":"CHANGELOG.md <span style='color:#111;'> 693B </span>","children":null,"spread":false},{"title":"include","children":[{"title":"tensorview","children":[{"title":"eigen_utils.h <span style='color:#111;'> 1.29KB </span>","children":null,"spread":false},{"title":"tensorview.h <span style='color:#111;'> 50.93KB </span>","children":null,"spread":false},{"title":"tools.h <span style='color:#111;'> 1.70KB </span>","children":null,"spread":false},{"title":"common.h <span style='color:#111;'> 4.80KB </span>","children":null,"spread":false},{"title":"pybind_utils.h <span style='color:#111;'> 4.50KB </span>","children":null,"spread":false},{"title":"mp_helper.h <span style='color:#111;'> 1.32KB </span>","children":null,"spread":false},{"title":"cuda_utils.h <span style='color:#111;'> 722B </span>","children":null,"spread":false},{"title":"torch_utils.h <span style='color:#111;'> 4.82KB </span>","children":null,"spread":false},{"title":"tensor.h <span style='color:#111;'> 30.83KB </span>","children":null,"spread":false},{"title":"prettyprint.h <span style='color:#111;'> 15.67KB </span>","children":null,"spread":false},{"title":"kernel_utils.h <span style='color:#111;'> 2.26KB </span>","children":null,"spread":false}],"spread":false},{"title":"tensorrt","children":[{"title":"inference.h <span style='color:#111;'> 6.79KB </span>","children":null,"spread":false}],"spread":false},{"title":"spconv","children":[{"title":"fused_spconv_ops.h <span style='color:#111;'> 4.58KB </span>","children":null,"spread":false},{"title":"nms_ops.h <span style='color:#111;'> 2.48KB </span>","children":null,"spread":false},{"title":"nms_functor.h <span style='color:#111;'> 1.24KB </span>","children":null,"spread":false},{"title":"indice.cu.h <span style='color:#111;'> 17.55KB </span>","children":null,"spread":false},{"title":"reordering.cu.h <span style='color:#111;'> 14.88KB </span>","children":null,"spread":false},{"title":"geometry.h <span style='color:#111;'> 5.19KB </span>","children":null,"spread":false},{"title":"pool_ops.h <span style='color:#111;'> 1.23KB </span>","children":null,"spread":false},{"title":"point2voxel.h <span style='color:#111;'> 9.97KB </span>","children":null,"spread":false},{"title":"pillar_scatter_ops.h <span style='color:#111;'> 2.06KB </span>","children":null,"spread":false},{"title":"maxpool.h <span style='color:#111;'> 1.63KB </span>","children":null,"spread":false},{"title":"cublas_gemm.h <span style='color:#111;'> 1.85KB </span>","children":null,"spread":false},{"title":"box_iou.h <span style='color:#111;'> 5.94KB </span>","children":null,"spread":false},{"title":"nms_gpu.h <span style='color:#111;'> 781B </span>","children":null,"spread":false},{"title":"pillar_scatter_functor.h <span style='color:#111;'> 1.02KB </span>","children":null,"spread":false},{"title":"spconv_ops.h <span style='color:#111;'> 2.53KB </span>","children":null,"spread":false},{"title":"reordering.h <span style='color:#111;'> 1.60KB </span>","children":null,"spread":false},{"title":"indice.h <span style='color:#111;'> 2.48KB </span>","children":null,"spread":false},{"title":"nms.h <span style='color:#111;'> 7.38KB </span>","children":null,"spread":false}],"spread":false},{"title":"cuhash","children":[{"title":"hash_table.h <span style='color:#111;'> 8.70KB </span>","children":null,"spread":false},{"title":"hash_table.cuh <span style='color:#111;'> 9.47KB </span>","children":null,"spread":false},{"title":"cuda_util.h <span style='color:#111;'> 2.70KB </span>","children":null,"spread":false},{"title":"debugging.h <span style='color:#111;'> 2.54KB </span>","children":null,"spread":false},{"title":"definitions.h <span style='color:#111;'> 3.68KB </span>","children":null,"spread":false},{"title":"hash_functions.h <span style='color:#111;'> 3.30KB </span>","children":null,"spread":false}],"spread":false},{"title":"utility","children":[{"title":"timer.h <span style='color:#111;'> 1.70KB </span>","children":null,"spread":false}],"spread":false},{"title":"torch_utils.h <span style='color:#111;'> 3.67KB </span>","children":null,"spread":false},{"title":"paramsgrid.h <span style='color:#111;'> 2.09KB </span>","children":null,"spread":false},{"title":"tsl","children":[{"title":"robin_map.h <span style='color:#111;'> 24.23KB </span>","children":null,"spread":false},{"title":"robin_growth_policy.h <span style='color:#111;'> 10.87KB </span>","children":null,"spread":false},{"title":"robin_hash.h <span style='color:#111;'> 44.27KB </span>","children":null,"spread":false}],"spread":false}],"spread":false}],"spread":false}],"spread":true}]

评论信息

免责申明

【只为小站】的资源来自网友分享,仅供学习研究,请务必在下载后24小时内给予删除,不得用于其他任何用途,否则后果自负。基于互联网的特殊性,【只为小站】 无法对用户传输的作品、信息、内容的权属或合法性、合规性、真实性、科学性、完整权、有效性等进行实质审查;无论 【只为小站】 经营者是否已进行审查,用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。
本站所有资源不代表本站的观点或立场,基于网友分享,根据中国法律《信息网络传播权保护条例》第二十二条之规定,若资源存在侵权或相关问题请联系本站客服人员,zhiweidada#qq.com,请把#换成@,本站将给予最大的支持与配合,做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明