search

tonylins / pytorch-mobilenet-v2

A PyTorch implementation of MobileNet V2 architecture and pretrained model.
Apache License 2.0
1.37k stars 328 forks source link

Model architecture #22

Closed TangAL0203 closed 5 years ago

TangAL0203 commented 5 years ago

why your implementation doesn't include avgpool layer? As the paper shows, mobilenet-v2 should has one pooling layer in the last!

tonylins commented 5 years ago

It's implemented in Line 108 in MobileNetV2.py