/pytorch_senet

A PyTorch implementation of SE-ResNet-preact

Primary LanguagePythonMIT LicenseMIT

PyTorch Implementation of SENet

Usage

$ python main.py --block_type basic --depth 110 --se_reduction 16 --outdir results

References

  • Hu, Jie, Li Shen, and Gang Sun. "Squeeze-and-excitation networks." arXiv preprint arXiv:1709.01507 (2017). arXiv:1709.01507