Update GroupNorm for 16 Groups

NVIDIA / apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

BSD 3-Clause "New" or "Revised" License

8.2k stars 1.36k forks source link

Closed alpha0422 closed 11 months ago

alpha0422 commented 11 months ago

This PR:

This PR is a must-have in 23.09 container.

Original crediet @Zhuoyao1012, commit b68096f.

alpha0422 commented 11 months ago

@crcrpar, could you take a review and see if there's any issue?