transmuteAI/trailmet

MobileNetv2 skip connections differing from torchvision implementation

Opened this issue a year ago · 0 comments

sydarb commented a year ago

There shouldn't be any skip connections for inverted residual blocks having input channel != output channel.
ref : https://github.com/pytorch/vision/blob/main/torchvision/models/mobilenetv2.py#L32

trailmet/trailmet/models/mobilenet.py

Line 74 in b595b98

if stride == 1 and in_planes != out_planes: