Yucao42 / examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

PyTorch Examples

A repository showcasing examples of using PyTorch

  • MNIST 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 - create an LSTM to learn Sine waves

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

About

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

License:BSD 3-Clause "New" or "Revised" License


Languages

Language:Python 81.9%Language:C++ 11.4%Language:Cuda 3.5%Language:CMake 2.2%Language:Shell 1.0%