About

A collection of various deep learning models, in TensorFlow2 and PyTorch.

Basic Models

Perceptron

• PyTorch : Github | Nbviewer
• TensorFlow : Github | Nbviewer

Logistic Regression

• PyTorch : Github | Nbviewer
• TensorFlow : Github | Nbviewer

Multilayer Perceptron

• PyTorch : Github | Nbviewer
• TensorFlow : Github | Nbviewer

Convolutional Neural Network (CNNs)

Convolutional Neural Network

• PyTorch : Github | Nbviewer
• TensorFlow : Github | Nbviewer

LeNet

• LeNet-5 on MNIST
  • PyTorch : Github | Nbviewer
• LeNet-5 on CIFAR-10
  • PyTorch : Github | Nbviewer

AlexNet

• AlexNet on CIFAR-10
  • PyTorch : Github | Nbviewer

VGG

• VGG-16 on CIFAR-10
  • PyTorch : Github | Nbviewer
• VGG-19 on CIFAR-10
  • PyTorch : Github | Nbviewer

ResNet

• ResNet-34 on MNIST
  • PyTorch : Github | Nbviewer
• ResNet-50 on MNIST
  • PyTorch : Github | Nbviewer
• ResNet-101 on CIFAR-10
  • PyTorch : Github | Nbviewer

Network in Network

• Network in Network on CIFAR-10
  • PyTorch : Github | Nbviewer

DenseNet

• DenseNet-121 on CIFAR-10
  • PyTorch : Github | Nbviewer

Transfer Learning

• Transfer Learning Example (VGG16 pre-trained on ImageNet for Cifar-10)
  • PyTorch : Github | Nbviewer

Recurrent Neural Networks (RNNs)

Many-to-one: Sentiment Analysis / Classification

• A simple single-layer RNN on IMDB
  • PyTorch : Github | Nbviewer
• RNN with LSTM cells on IMDB
  • PyTorch : Github | Nbviewer
• RNN with GRU cells on IMDB
  • PyTorch : Github | Nbviewer
• RNN with LSTM cells on IMDB and pre-trained GloVe word vectors
  • PyTorch : Github | Nbviewer
• Multilayer bi-directional RNN on IMDB
  • PyTorch : Github | Nbviewer

Many-to-Many / Sequence-to-Sequence

• A simple character RNN to generate new text (Charles Dickens)
  • PyTorch : Github | Nbviewer

Autoencoders

Fully-connected Autoencoders

• Autoencoder (MNIST)
  • PyTorch : Github | Nbviewer

Variational Autoencoders

• Variational Autoencoder
  • PyTorch : Github | Nbviewer

Generative Adversarial Networks (GANs)

Fully Connected GAN on MNIST

• PyTorch : Github | Nbviewer

Graph Neural Networks (GNNs)

• Graph Neural Network
  • PyTorch : Github | Nbviewer