MobileNetV2 Width_mult

[CowbungaMk1]

Hello, I wanted to try different width mobileNets based on their paper, however I get an error that the weight size does not match the number of channels. For example, width_mult = 2.0 gives:

RuntimeError: Given groups=1, weight of size [48, 24, 1, 1], expected input[16, 48, 160, 221] to have 24 channels, but got 48 channels instead

Ive tried it with others (other then 1.0) but similar errors. Not sure where the mismatch is occurring.

Thanks