Branch: master
Find file History
Permalink
Type Name Latest commit message Commit time
..
Failed to load latest commit information.
pytorch_ipynb sequential example with hooks Feb 20, 2019
tensorflow_ipynb vgg16 in pytorch Feb 2, 2018
README.md sequential example with hooks Feb 20, 2019

README.md

Python 3.6

Model Zoo

A collection of standalone TensorFlow and PyTorch models in Jupyter Notebooks.

Classifiers

Convolutional Classifiers

Basic

  • Convolutional Neural Network [TensorFlow] [PyTorch]
  • Convolutional Neural Network with He Initialization [PyTorch]
  • All-Convolutional Neural Network [PyTorch]

VGG

  • Convolutional Neural Network VGG-16 [TensorFlow] [PyTorch]
  • VGG-16 Gender Classifier Trained on CelebA [PyTorch]
  • Convolutional Neural Network VGG-19 [PyTorch]

ResNet

  • Convolutional ResNet and Residual Blocks [PyTorch]
  • ResNet-18 Digit Classifier Trained on MNIST [PyTorch]
  • ResNet-18 Gender Classifier Trained on CelebA [PyTorch]
  • ResNet-34 Digit Classifier Trained on MNIST [PyTorch]
  • ResNet-34 Gender Classifier Trained on CelebA [PyTorch]
  • ResNet-50 Digit Classifier Trained on MNIST [PyTorch]
  • ResNet-50 Gender Classifier Trained on CelebA [PyTorch]
  • ResNet-101 Gender Classifier Trained on CelebA [PyTorch]
  • ResNet-152 Gender Classifier Trained on CelebA [PyTorch]

Metric Learning

  • Siamese Network with Multilayer Perceptrons [TensorFlow]

Autoencoders

Regular Autoencoders

  • Autoencoder [TensorFlow] [PyTorch]
  • Convolutional Autoencoder with Deconvolutions [TensorFlow] [PyTorch]
  • Convolutional Autoencoder with Deconvolutions (without pooling operations) [PyTorch]
  • Convolutional Autoencoder with Nearest-neighbor Interpolation [TensorFlow] [PyTorch]
  • Convolutional Autoencoder with Nearest-neighbor Interpolation -- Trained on CelebA [PyTorch]
  • Convolutional Autoencoder with Nearest-neighbor Interpolation -- Trained on Quickdraw [PyTorch]

Variational Autoencoders

  • Variational Autoencoder [PyTorch]
  • Convolutional Variational Autoencoder [PyTorch]

Conditional Variational Autoencoders

  • Conditional Variational Autoencoder (with labels in reconstruction loss) [PyTorch]
  • Conditional Variational Autoencoder (without labels in reconstruction loss) [PyTorch]
  • Convolutional Conditional Variational Autoencoder (with labels in reconstruction loss) [PyTorch]
  • Convolutional Conditional Variational Autoencoder (without labels in reconstruction loss) [PyTorch]

General Adversarial Networks

Tips and Tricks

PyTorch Workflows

Datasets

Parallel Computing

Other

TensorFlow Workflows