search

pytorch / vision

Datasets, Transforms and Models specific to Computer Vision
https://pytorch.org/vision
BSD 3-Clause "New" or "Revised" License
15.99k stars 6.92k forks source link

Imagenet Pre-trained model for other Depth Multiplier #2951

Open NaifahNurya opened 3 years ago

NaifahNurya commented 3 years ago

On the mnasnet model under mnasnet.py file, the link provided for imagenet pretrained model is only for two depth multiplier, as shown in the code below:

_MODEL_URLS = { "mnasnet0_5": "https://download.pytorch.org/models/mnasnet0.5_top1_67.823-3ffadce67e.pth", "mnasnet0_75": None, "mnasnet1_0": "https://download.pytorch.org/models/mnasnet1.0_top1_73.512-f206786ef8.pth", "mnasnet1_3": None }

Can you provide the link for Imagenet pre-trained model for mnasnet0_75 and mnasnet1_3?

NaifahNurya commented 3 years ago

Any help/updates on this request @vfdev-5