├── .gitignore ├── LICENSE ├── README.md ├── requirements.txt ├── setup.py ├── test.py └── torchsyncbn ├── __init__.py ├── functional ├── __init__.py ├── _syncbn │ ├── __init__.py │ ├── _ext │ │ └── __init__.py │ └── src │ │ ├── common.h │ │ ├── syncbn.cpp │ │ ├── syncbn.cu │ │ ├── syncbn.cu.h │ │ └── syncbn.h └── syncbn.py └── nn ├── __init__.py └── syncbn.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shinsumicco/pytorch-syncbn/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shinsumicco/pytorch-syncbn/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shinsumicco/pytorch-syncbn/HEAD/README.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | cffi 2 | future 3 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shinsumicco/pytorch-syncbn/HEAD/setup.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shinsumicco/pytorch-syncbn/HEAD/test.py -------------------------------------------------------------------------------- /torchsyncbn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shinsumicco/pytorch-syncbn/HEAD/torchsyncbn/__init__.py -------------------------------------------------------------------------------- /torchsyncbn/functional/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shinsumicco/pytorch-syncbn/HEAD/torchsyncbn/functional/__init__.py -------------------------------------------------------------------------------- /torchsyncbn/functional/_syncbn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shinsumicco/pytorch-syncbn/HEAD/torchsyncbn/functional/_syncbn/__init__.py -------------------------------------------------------------------------------- /torchsyncbn/functional/_syncbn/_ext/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shinsumicco/pytorch-syncbn/HEAD/torchsyncbn/functional/_syncbn/_ext/__init__.py -------------------------------------------------------------------------------- /torchsyncbn/functional/_syncbn/src/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shinsumicco/pytorch-syncbn/HEAD/torchsyncbn/functional/_syncbn/src/common.h -------------------------------------------------------------------------------- /torchsyncbn/functional/_syncbn/src/syncbn.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shinsumicco/pytorch-syncbn/HEAD/torchsyncbn/functional/_syncbn/src/syncbn.cpp -------------------------------------------------------------------------------- /torchsyncbn/functional/_syncbn/src/syncbn.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shinsumicco/pytorch-syncbn/HEAD/torchsyncbn/functional/_syncbn/src/syncbn.cu -------------------------------------------------------------------------------- /torchsyncbn/functional/_syncbn/src/syncbn.cu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shinsumicco/pytorch-syncbn/HEAD/torchsyncbn/functional/_syncbn/src/syncbn.cu.h -------------------------------------------------------------------------------- /torchsyncbn/functional/_syncbn/src/syncbn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shinsumicco/pytorch-syncbn/HEAD/torchsyncbn/functional/_syncbn/src/syncbn.h -------------------------------------------------------------------------------- /torchsyncbn/functional/syncbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shinsumicco/pytorch-syncbn/HEAD/torchsyncbn/functional/syncbn.py -------------------------------------------------------------------------------- /torchsyncbn/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shinsumicco/pytorch-syncbn/HEAD/torchsyncbn/nn/__init__.py -------------------------------------------------------------------------------- /torchsyncbn/nn/syncbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/shinsumicco/pytorch-syncbn/HEAD/torchsyncbn/nn/syncbn.py --------------------------------------------------------------------------------