-
Notifications
You must be signed in to change notification settings - Fork 388
Open
Description
What are these levels?
SSD/ssd/modeling/backbone/mobilenetv3.py
Lines 177 to 182 in 68dc0a2
| self.extras = nn.ModuleList([ | |
| InvertedResidual(960, _make_divisible(960 * 0.2, 8), 512, 3, 2, True, True), | |
| InvertedResidual(512, _make_divisible(512 * 0.25, 8), 256, 3, 2, True, True), | |
| InvertedResidual(256, _make_divisible(256 * 0.5, 8), 256, 3, 2, True, True), | |
| InvertedResidual(256, _make_divisible(256 * 0.25, 8), 64, 3, 2, True, True), | |
| ]) |
and these
SSD/ssd/modeling/backbone/mobilenet.py
Lines 89 to 94 in 68dc0a2
| self.extras = nn.ModuleList([ | |
| InvertedResidual(1280, 512, 2, 0.2), | |
| InvertedResidual(512, 256, 2, 0.25), | |
| InvertedResidual(256, 256, 2, 0.5), | |
| InvertedResidual(256, 64, 2, 0.25) | |
| ]) |
I planning integrate mobilenet v4 from https://github.com/jaiwei98/MobileNetV4-pytorch/blob/main/mobilenet/mobilenetv4.py or https://huggingface.co/blog/rwightman/mobilenetv4
Should I add extra levels like in v3? Replacing InvertedResidual with UniversalInvertedResidual
Metadata
Metadata
Assignees
Labels
No labels