pretrained model doesn't match model

linksense / LightNetPlusPlus

LightNet++: Boosted Light-weighted Networks for Real-time Semantic Segmentation

https://ansleliu.github.io/LightNet.html

MIT License

243 stars 54 forks source link

pretrained model doesn't match model #1

Closed harryhan618 closed 5 years ago

harryhan618 commented 5 years ago

I tried 'MobileNetV2Plus X1.0'. But its OrderedDict keys don't match model's keys. I'm confused.

ansleliu commented 5 years ago

@harryhan618 please, modify mobilenetv2plus: self.aspp_scse = nn.Sequential(OrderedDict([("norm", norm_act(aspp_chns)), ("scse", PBCSABlock(in_chns=aspp_chns, reduct_ratio=16, dilation=16, use_res=True))]))