facebookresearch / SparseConvNet

Submanifold sparse convolutional networks
https://github.com/facebookresearch/SparseConvNet
Other
2.04k stars 332 forks source link

Group normalization with sparse convolution #200

Closed victoryc closed 1 year ago

victoryc commented 3 years ago

I see that SparseConvNet has implementation for batch normalization. Is it possible to use group normalization with SparseConvNet?

My specific use case is the following. As of now, I am using the U-Net class available in SparseConvNet. But, it uses batch normalization. I would like try out group normalization instead.

Thanks.

btgraham commented 1 year ago

Added as sparseconvnet.SparseGroupNorm