Skip to content
This repository has been archived by the owner on Apr 1, 2021. It is now read-only.

[Feature Request] Add Squeeze and Excitation block #11

Open
wants to merge 4 commits into
base: master
Choose a base branch
from
Open

[Feature Request] Add Squeeze and Excitation block #11

wants to merge 4 commits into from

Conversation

bilal2vec
Copy link

Hi,

I added the squeeze and excitation block from the Squeeze-and-Excitation Networks paper.

The code for the SE block is in torchcontrib/nn/modules/linear.py and the tests for it are in test/test_nn.py

@bilal2vec bilal2vec changed the title Add Squeeze and Excitation block [Feature Request] Add Squeeze and Excitation block Oct 15, 2018
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant