search

xiaolai-sqlai / mobilenetv3

mobilenetv3 with pytorch，provide pre-train model
MIT License
1.6k stars 340 forks source link

SeModule Error #50

Open wyyaa123 opened 8 months ago

wyyaa123 commented 8 months ago

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