PyTorch Hub
For Researchers

classify birds using this fine-grained image classifier

DeepLabV3 model with a ResNet-101 backbone

Transformer models for English-French and English-German translation.

WaveGlow model for generating speech from mel spectrograms (generated by Tacotron2)

ResNext models trained with billion scale weakly-supervised data.

A simple generative image model for 64x64 images

High-quality image generation of fashion, celebrity faces

ResNet and ResNext models introduced in the "Billion scale semi-supervised learning for image classification" paper

PyTorch implementations of popular NLP Transformers

U-Net with batch normalization for biomedical image segmentation with pretrained weights for abnormality segmentation in brain MRI

Single Shot MultiBox Detector model for object detection

The Tacotron 2 model for generating mel spectrograms from text