[BUG] Skip connection should be always added to DW branches.

[grygielski]

Hi @shoutOutYangJie, I've noticed that you have a bug in your MobileOne implementation:

MobileOne/mobileone.py

Line 77 in 48ca6c9

self.dw_bn_layer = nn.BatchNorm2d(in_channels) if out_channels == in_channels and stride == 1 else None

You are adding DW BN layer based on in_channels == out_channels condition but it should be always added because for DW part input channels are always equal to output channels. This condition should be only checked for PW part as there might be channel change.

[shoutOutYangJie]

Oh, yes, thank you. I will refer your issue to ReadMe. Because I don't plan to retrain the model, so I don't want to rewrite it.
Thank you again.

[shoutOutYangJie]

and the official mobileone source code has been opened.