├── LICENSE ├── README.md ├── arch.png ├── module ├── backbone.py └── stoch_norm.py ├── requirements.txt ├── train.ipynb ├── train.py └── utils ├── tools.py └── transforms.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/StochNorm/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/StochNorm/HEAD/README.md -------------------------------------------------------------------------------- /arch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/StochNorm/HEAD/arch.png -------------------------------------------------------------------------------- /module/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/StochNorm/HEAD/module/backbone.py -------------------------------------------------------------------------------- /module/stoch_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/StochNorm/HEAD/module/stoch_norm.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/StochNorm/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/StochNorm/HEAD/train.ipynb -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/StochNorm/HEAD/train.py -------------------------------------------------------------------------------- /utils/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/StochNorm/HEAD/utils/tools.py -------------------------------------------------------------------------------- /utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/thuml/StochNorm/HEAD/utils/transforms.py --------------------------------------------------------------------------------