Skip to content

varunkashyapks/fastestimator

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

71 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

FastEstimator

Build Status

FastEstimator is a high-level deep learning API. With the help of FastEstimator, you can easily build a high-performance deep learning model and run it anywhere. 😉

Prerequisites:

  • Python3
  • TensorFlow 1.12.0
  • Horovod (Only needed for distributed training)

Installation

pip install fastestimator

Docker

Docker container creates isolated virtual environment that shares resources with host machine. Docker provides an easy way to set up FastEstimator running environment, users can either build from dockerfile or pull from Docker Hub.

Build Image from Dockerfile

  • GPU: docker build -t fastestimator_gpu -f docker/Dockerfile.gpu .
  • CPU: docker build -t fastestimator_cpu -f docker/Dockerfile.cpu .

Pull Image from Docker Hub

  • GPU: docker pull fastestimator/fastestimator:latest-gpu

Running your first FastEstimator training

  • macOS/Linux:
$ fastestimator train --entry_point tutorial/mnist.py
  • Windows, macOS and Linux:

    check out this notebook

Tests

$ ./fastestimator/test/tests.sh

Documentation

For more info about building models and training, check out FastEstimator documentation

License

Apache License 2.0

Packages

No packages published

Languages

  • Python 99.9%
  • Shell 0.1%