search

zhaoyuzhi / PyTorch-MobileNet-v123

Provide implementations and pre-trained models of MobileNet-v1, v2, and v3
15 stars 6 forks source link

pre-trained-models #1

Open Jonathan-fzq opened 2 years ago

Jonathan-fzq commented 2 years ago

What dataset is the model pre-trained on？

zhaoyuzhi commented 2 years ago

imagenet

https://github.com/zhaoyuzhi/PyTorch-MobileNet-v123/blob/e5979da4689d2813b50694e8e51051c916a051c4/train.py#L28

Jonathan-fzq commented 2 years ago

thank you!