GitPedia
mit-han-lab

mit-han-lab/torchsparse

[MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.

5 Releases
Latest: 3y ago
v2.0.0Latest
kentang-mitkentang-mit·3y ago·June 18, 2023
GitHub

TorchSparse v2.0 (MLSys 2022 version).

v1.4.0
zhijian-liuzhijian-liu·4y ago·June 25, 2021
GitHub

New Features

  • Supported mixed-precision training and inference with `torch.cuda.amp` (https://github.com/mit-han-lab/torchsparse/pull/69, https://github.com/mit-han-lab/torchsparse/pull/75).
  • Added generalized sparse convolution (https://github.com/mit-han-lab/torchsparse/pull/77).
  • Added group normalization (https://github.com/mit-han-lab/torchsparse/pull/63).

📋 API Changes

  • `sparse_{collate,quantize}` now needs to be imported from `torchsparse.utils.{collate,quantize}`.
  • `sparse_quantize` now takes in `coords`, `voxel_size` (defaults to `1.0`), `return_index` (defaults to `False`) and `return_inverse` (defaults to `False`) as input, and returns the quantized `coords` as well as `indices` and `inverse_indices` (if requested).
  • `transpose` is renamed as `transposed` for `torchsparse.nn.Conv3d` and `torchsparse.nn.functional.conv3d`.
  • `KernelRegion` is removed and replaced with `get_kernel_offsets` (in `torchsparse.nn.utils`).
v1.2.0
zhijian-liuzhijian-liu·5y ago·June 2, 2021
GitHub
v1.1.0
zhijian-liuzhijian-liu·5y ago·March 20, 2021
GitHub
v1.0.0
zhijian-liuzhijian-liu·5y ago·September 17, 2020
GitHub

v1.0.0