search

NVIDIA / MinkowskiEngine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors
https://nvidia.github.io/MinkowskiEngine
Other
2.43k stars 360 forks source link

Any plan on supporting half precision / mixed precision? #486

Open HaoyiZhu opened 2 years ago

HaoyiZhu commented 2 years ago

Hi, thank for the wonderful repo. Do you have any plan to support half or mixed precision training?

I think this is very useful when using high dimensional data and large models as it can greatly save memory.

I noticed that others have asked similar issues before but no reply yet.

Thanks!