search

pytorch / serve

Serve, optimize and scale PyTorch models in production
https://pytorch.org/serve/
Apache License 2.0
4.19k stars 858 forks source link

Add more torchserve examples using popular PyTorch models #649

Closed prashantsail closed 3 years ago

prashantsail commented 4 years ago

Description

To enrich and diversify the torchserve examples, we would like to add a few more by using some of the popular PyTorch models.

I have segregated a small list -

Model Github Stars PR Overview
RoBERTa (fairseq) 8.9k #592 Models for translation, summarization, language modeling and other text generation tasks
SSD 4.5k Object Detection (Image size 300 x 300) [performance]
Tacotron 2 4.5k - Text to speech [performance]
Example already exists
3D-ResNets 2.4k Human action recognition in videos
ResNeSt 2.2k Split-Attention Network, A New ResNet Variant [performance]
CRAFT 1.5k Effectively detect text area in an image
DCGAN on FashionGen 1.1k #659 Create Images from input
Progressive Growing of GANs 1.1k - Create Images from input
Similar to DCGAN example
ProxylessNAS 1.1k Specialize CNN architectures for different hardware platforms (CPU \ GPU \ Mobile) [performance]
GhostNet 1.0k More Features from Cheap Operations [performance]

Most of it is sourced from PyTorch Hub

@dhaniram-kshirsagar @chauhang @maaquib @MengMeng96 @HamidShojanazeri @htappen Please provide your inputs on this ticket.

We are also open to any other suggestions

msaroufim commented 3 years ago

Closing because we already have many examples - I think it makes more sense to refactor our existing docs to make it trivial for people to add their own handlers