Deep learning Application framework


Keywords
fastestimator tensorflow
License
Apache-2.0
Install
pip install fastestimator-nightly==1.0b2.dev202001211040

Documentation

FastEstimator

Build Status 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:

  • Python >= 3.5

  • TensorFlow2

    • GPU: pip install tensorflow-gpu==2.0.0
    • CPU: pip install tensorflow==2.0.0

Installation

pip install fastestimator==1.0b0

Docker Hub

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

  • GPU: docker pull fastestimator/fastestimator:1.0b0-gpu
  • CPU: docker pull fastestimator/fastestimator:1.0b0-cpu

Start your first FastEstimator training

$ python ./apphub/image_classification/lenet_mnist/lenet_mnist.py

Documentation

For more info, check out FastEstimator documentation

Citation

Please cite FastEstimator in your publications if it helps your research:

@misc{dong2019fastestimator,
    title={FastEstimator: A Deep Learning Library for Fast Prototyping and Productization},
    author={Xiaomeng Dong and Junpyo Hong and Hsi-Ming Chang and Michael Potter and Aritra Chowdhury and
            Purujit Bahl and Vivek Soni and Yun-Chan Tsai and Rajesh Tamada and Gaurav Kumar and Caroline Favart and
            V. Ratna Saripalli and Gopal Avinash},
    year={2019},
    eprint={1910.04875},
    archivePrefix={arXiv},
    primaryClass={cs.CV}
}

License

Apache License 2.0