登录 注册
当前位置:主页 > 资源下载 > 46 > 在PyTorch中实现的空间稀疏卷积库spconv

在PyTorch中实现的空间稀疏卷积库spconv

  • 更新:2024-12-12 15:21:02
  • 大小:256KB
  • 推荐:★★★★★
  • 来源:网友上传分享
  • 类别:其它 - 开发技术
  • 格式:ZIP

资源介绍

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”的标志。 此项