PremBharwani/model-zoo

Implementations of various Deep Learning models in PyTorch and TensorFlow.

Model Zoo

This repository contains implementations of various deep learning research papers.

The models are broadly categorised into the folders Generative (e.g. various generative models), NLP (e.g. various recurrent neural networks (RNNs) and natural language processing (NLP) models), Classification (e.g. various CNN models to classify images), Object Detection, Multimodal , Super resolution , 3D Computer Vision. See the READMEs of respective models for more information.

Model Listing

• Generative models
  • AAE PyTorch
  • ACGAN PyTorch
  • CycleGAN PyTorch
  • DCGAN TensorFlow PyTorch
  • DiscoGAN PyTorch
  • InfoGAN TensorFlow PyTorch
  • MoCoGAN PyTorch
  • StarGAN PyTorch
  • VAEGAN PyTorch
  • VanillaGAN TensorFlow PyTorch
  • WGAN TensorFlow
  • cGAN TensorFlow PyTorch
  • SS-GAN TensorFlow
• NLP
  • BERT TensorFlow PyTorch
  • BI-LSTM_CRF TensorFlow
  • GloVe TensorFlow PyTorch
  • Transformer TensorFlow
  • Word2Vec PyTorch
• classification
  • ResNet PyTorch
  • T3D ConvNets TensorFlow
• Multimodal models
  • StackGAN PyTorch
  • VQA TensorFlow
  • WaveGAN TensorFlow
  • ImageCaption TensorFlow PyTorch
• Object detection
  • YOLOv2 TensorFlow
  • YOLOv3 TensorFlow
• Super resolution
  • ContextEncoder TensorFlow PyTorch
  • SRGAN TensorFlow PyTorch
  • SRCNN TensorFlow
  • VDSR PyTorch
• 3D Computer Vision
  • 3DGAN PyTorch