pretrained-models.pytorch icon indicating copy to clipboard operation
pretrained-models.pytorch copied to clipboard

Any plan for other network backbones??

Open chibai opened this issue 6 years ago • 1 comments

This is just advice. In production environment, latency is also very important. Do you have any plan to train small network, such as mobilenet v1&v2, shufflenet v1&v2 and so on???

chibai avatar Apr 16 '19 12:04 chibai

mobilenet, shufflenet, .... are already implemented and pretrained. they are available via torchvision models: https://pytorch.org/docs/stable/torchvision/models.html

imirzadeh avatar Aug 15 '19 23:08 imirzadeh