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

the dropout rate should be 0.2 in original paper? #12

Closed linrongc closed 6 years ago

linrongc commented 6 years ago

https://github.com/tensorflow/models/blob/master/research/slim/nets/mobilenet/mobilenet.py

tonylins commented 6 years ago

Hi,

Thanks. The original paper does not mention the dropout rate. But setting dropout rate to 0.5 is indeed my mistake, which makes the training much longer. I'll leave a notice in README.