anguoyang/pytorch-model-zoo

A collection of deep learning models implemented in PyTorch

Deep learning models in PyTorch [IN PROGRESS]

This repository contains deep learning models built in PyTorch. Intended for learning PyTorch, this repo is made understandable for someone with basic python and deep learning knowledge. All kinds of important deep learning models are implemented here. Links to the papers are also given.

The structure for each project is consistent:

• model.py - Contains the model of the neural network.
• train.py - Preprocessing the data and training the network.
• test.py - Infering from the trained network.
• any other supporting files.

Common folders:

• Trained folder - Trained models can will be saved here.
• Datasets folder - Datasets downloaded will be stored here.

Best practices like PEP8, dropout, batch normalization, suitable optimizers are used.

My trained models can be found here.

IMP: For training on the CPU remove '.cuda()' wherever you find it.

Models

1. CNNs

• Image Classification: Microsoft-ResNet 2015
• Spatial Transformation Network STN 2016
• You Only Look Once YOLO 2016
• Super resolution IEEE 2015
• Artistic Style Transfer Gatys 2015
• Deep Photo Style Transfer 2017
• Neural Doodle 2016
• Image Colorization UC Berkeley 2016

2. RNNs

• Speech Recognition: Deep Speech 2 2015 (If possible with attention) (Maybe Deep Voice )
• Generating sequences: LSTM 2013
• Sequence to sequence with attention: Text Summarizer
• Neural Machine Translation: Google 2016
• Conversational Model: Google 2015
• Skip Thoughts sentence to vector

3. GANs

• Deep Convolutional GAN: DCGAN 2015
• Text to image synthesis 2016

4. Reinforcement

• Deep Q Learning Deep Mind 2015
• Asynchronous advantage actor-critic A3C

5. Others

• Image Captioning with attention Bengio 2016
• Hybrid Computing with a NN and external memory Nature 2016
• Network in Network 2014

References:

• https://github.com/songrotek/Deep-Learning-Papers-Reading-Roadmap
• https://github.com/terryum/awesome-deep-learning-papers
• https://github.com/yunjey/pytorch-tutorial
• https://github.com/pytorch/examples
• https://github.com/bharathgs/Awesome-pytorch-list
• https://github.com/ritchieng/the-incredible-pytorch