PyTorch Examples

WARNING: if you fork this repo, github actions will run daily on it. To disable this, go to /examples/settings/actions and Disable Actions for this repository

A repository showcasing examples of using PyTorch

• Image classification (MNIST) using Convnets
• Word level Language Modeling using LSTM RNNs
• Training Imagenet Classifiers with Residual Networks
• Generative Adversarial Networks (DCGAN)
• Variational Auto-Encoders
• Superresolution using an efficient sub-pixel convolutional neural network
• Hogwild training of shared ConvNets across multiple processes on MNIST
• Training a CartPole to balance in OpenAI Gym with actor-critic
• Natural Language Inference (SNLI) with GloVe vectors, LSTMs, and torchtext
• Time sequence prediction - use an LSTM to learn Sine waves
• Implement the Neural Style Transfer algorithm on images
• Several examples illustrating the C++ Frontend

Additionally, a list of good examples hosted in their own repositories:

• Neural Machine Translation using sequence-to-sequence RNN with attention (OpenNMT)