search

BangguWu / ECANet

Code for ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks
MIT License
1.27k stars 198 forks source link

How to implement ECA-NS by PyTorch? #13

Closed splinter21 closed 4 years ago

BangguWu commented 4 years ago

Use nn.functional.unfold() and conv1d with inchannel=C, outchannel=C and group=C.