search

wofmanaf / SA-Net

Code for our ICASSP 2021 paper: SA-Net: Shuffle Attention for Deep Convolutional Neural Networks
246 stars 42 forks source link

Getting error #16

Open DeviSPhd opened 2 years ago

DeviSPhd commented 2 years ago

when using sa_layer in deeplearning model getting following error, kindly help to me, solve this x = x.reshape(b * self.groups, -1, h, w) RuntimeError: shape '[64, -1, 64, 64]' is invalid for input of size 327680 kindly solve this

wofmanaf commented 2 years ago

这个应该是分组的组数不能被channel维度整除造成的，可以尝试较小的分组数

DeviSPhd commented 2 years ago

Thank you very much for your response

On Fri, 12 Nov 2021, 1:10 pm zhql, @.***> wrote:

这个应该是分组的组数不能被channel维度整除造成的，可以尝试较小的分组数

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub https://github.com/wofmanaf/SA-Net/issues/16#issuecomment-966890144, or unsubscribe https://github.com/notifications/unsubscribe-auth/ATPSOBBJOTDSRMQGCO23IS3ULTAIJANCNFSM5HPQ3UVA . Triage notifications on the go with GitHub Mobile for iOS https://apps.apple.com/app/apple-store/id1477376905?ct=notification-email&mt=8&pt=524675 or Android https://play.google.com/store/apps/details?id=com.github.android&referrer=utm_campaign%3Dnotification-email%26utm_medium%3Demail%26utm_source%3Dgithub.