Giters

VainF / DeepLabV3Plus-Pytorch

Pretrained DeepLabv3 and DeepLabv3+ for Pascal VOC & Cityscapes

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

MobileNetV2 Width_mult

CowbungaMk1 opened this issue · comments

commented

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