Skip to content

openAGI/models

Repository files navigation

Examples using tefla.

  1. Imagnet classification using Inception resnet v2
  2. Variational autoencoder training
  3. Spatial transformer network traning.
  4. Unrolled GAN implementation
  5. LSTM RNN example
  6. Common datasets reading examples.
  7. Autonencoder using Gumbel Softmax
  8. Classification networks, Inception resnet, SENET, Xception, RESNET

Pretrained Weights

  1. VGG19 Trained on ImageNet
  2. Inception_Resnet_v2 trained on ImageNet

Requirements

[GPU version]
pip install tensorflow-gpu>=1.8.0
[CPU version]
pip install tensorflow>=1.8.0

pip install tefla

Train a NLP model on the IMDB dataset

[models as CWD]
export PYTHONPATH=$PYTHONPATH:`$PWD`
python nlp/lstm_imdb.py

About

Tefla implementations of deep learning models

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published