taokong / group_normalization

pytorch implementation of group normalization in https://arxiv.org/abs/1803.08494

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

group_normalization

GN has been implemented by official PyTorch, so I recommend to use the official one which is more memory efficient.

pytorch re-implementation of group normalization in https://arxiv.org/abs/1803.08494

ref:

@article{wu2018group,
    title={Group Normalization},
    author={Yuxin Wu, Kaiming He},
    journal={arXiv preprint arXiv:1803.08494},
    year={2018}

About

pytorch implementation of group normalization in https://arxiv.org/abs/1803.08494


Languages

Language:Python 100.0%