d-li14 / mobilenetv3.pytorch

74.3% MobileNetV3-Large and 67.2% MobileNetV3-Small model on ImageNet

Home Page:https://arxiv.org/abs/1905.02244

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

not able to replicate the results

dipendra009 opened this issue · comments

Hi,
I am running your implementation with the hyper parameters you provided using the examples/imagenet/main.py provided by PyTorch. Seems like the accuracy metrics are stuck around after epoch 20. Could you please share your training script with hyper parameter settings?

Please refer to https://github.com/d-li14/mobilenetv2.pytorch/blob/master/imagenet.py for the training script. Also, the training curve with cosine LR decay should look like this https://github.com/d-li14/ghostnet.pytorch/raw/master/figs/ablation.png, in which the accuracy grows slowly in the middle, but is not stuck.