Code Monkey home page Code Monkey logo

tefla's Introduction

Tefla

Tefla is a deep learning mini-framework that sits on top of Tensorflow. Tefla's primary goal is to enable simple, stable, end-to-end deep learning. This means that Tefla supports:

  • Data setup
  • Batch preprocessing and data layout.
  • Training
  • A model definition DSL.
  • A training config DSL.
  • Data loading with data-augmentation and rebalancing.
  • Training with support for visualization, logging, custom metrics, and most importantly - resumption of training from an earlier epoch with a new learning rate.
  • Pluggable learning rate decay policies.
  • Stability and solidity - which translates to days and weeks of training without memory blowup and epoch time degradations.
  • Tensorboard visualization of epoch metrics, augmented images, model graphs, and layer activations, weights and gradients.
  • Prediction (with ensembling via mean score or voting).
  • Metrics on prediction outputs.
  • First class support for transfer learning and fine-tuning based on vgg16, resnet50, resnet101, and resnet152.
  • Serving of models via a REST API (coming soon).

Tefla contains command line scripts to do batch preprocessing, training, prediction, and metrics, thus supporting a simple yet powerful deep learning workflow.

Documentation is coming soon. For now, the mnist example(s) can help you to get started.

Tefla is very much a work in progress. Contributions are welcome!

An interesting fork of tefla is available here: www.github.com/n3011/tefla. Both projects are evolving independently, with a cross-pollination of ideas.

tefla's People

Contributors

litan avatar siddhantkapil avatar n3011 avatar shivam-kotwalia avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.