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