BangguWu / ECANet

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

change channel condition for kernal size #32

Closed yayalll closed 4 years ago

yayalll commented 4 years ago

When I load the eca_mobilenetv2_k13.pth 2.tar, it told me that the defined model ECA_MobileNetV2 and the pertained model don't match correctly in feature 11,12,13 layers. And when I change "<=" into "=", it worked.