mobilenetv3 with pytorch，provide pre-train model
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool
wyyaa123 opened this issue 10 months ago · comments
你好，在SeModule中插入的的BatchNorm2d模块是不是错了， 因为在AdaptiveAvgPool2d后特征图大小变为1x1，对这种特征图进行BN的话会报错。我的操作系统是Ubuntu18.04，torch==1.11.0+cu113; torchvision==0.12.0+cu113; CUDA:11.3
