add torch.device support for quantization

NVIDIA / MinkowskiEngine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

https://nvidia.github.io/MinkowskiEngine

Other

2.47k stars 367 forks source link

Closed eamonn-zh closed 1 year ago

eamonn-zh commented 1 year ago

add torch.device support for the ME.utils.sparse_quantize() function