👩💻 Join our community of thousands of amazing developers!
senet.pytorch PyTorch implementation of SENet An implementation of SENet, proposed in Squeeze-and-Excitation Networks by Jie Hu, Li Shen and Gang Sun, who are the winners of ILSVRC 2017 classification competition. Now SE-ResNet (18, 34, 50, 101, 152/20, 32) and SE-Inception-v3 are implemented. python cifar.py runs SE-ResNet20 with Cifar10 dataset. python imagenet.py and python -m torch.distributed.launch --nproc_per_node=${NUM_GPUS} imagenet.py run SE-ResNet50 with ImageNet(2012) dataset, Y...