incorrect network structure

[Pi]

Hi @d-li14

Thank you for your work.
When I read the original MobileNetV3 paper and compared with your code, I noticed a discrepancy in the code. Specifically, in this line [https://github.com/d-li14/mobilenetv3.pytorch/blob/master/mobilenetv3.py#L211], the code is: [5, 672, 160, 1, 1, 2].
However, the corresponding line in table 1 of the original paper is: 7
2 × 160, bneck, 5x5 960-160-checkmark-HS-1 (row 6 counted from the bottom)

Can you help to confirm that and re-adjust your code accordingly?

Thanks,

P

[Pi]

Hi @d-li14 ,

I figured out the problem.
I read the v2 of the paper and your code is based on v1 of the paper (Searching for MobileNetV3).

Thanks,

P