Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

SeModule Error #50

Open
wyyaa123 opened this issue Oct 21, 2023 · 1 comment
Open

SeModule Error #50

wyyaa123 opened this issue Oct 21, 2023 · 1 comment

Comments

@wyyaa123
Copy link

你好，在SeModule中插入的的BatchNorm2d模块是不是错了， 因为在AdaptiveAvgPool2d后特征图大小变为1x1，对这种特征图进行BN的话会报错。我的操作系统是Ubuntu18.04，torch==1.11.0+cu113; torchvision==0.12.0+cu113; CUDA:11.3
@ffxxy
Copy link

ffxxy commented Sep 19, 2024

我也是报错了，把BN去掉就好了，我记得好像原始Mobilenetv3的SE也没有用BN吧

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@wyyaa123 @ffxxy