Edge Machine Learning: Tensorflow Library

This directory includes, Tensorflow implementations of various techniques and algorithms developed as part of EdgeML. Currently, the following algorithms are available in Tensorflow:

1. Bonsai
2. EMI-RNN
3. FastRNN & FastGRNN
4. ProtoNN

The TensorFlow compute graphs for these algoriths are packaged as edgeml.graph. Trainers for these algorithms are in edgeml.trainer. Usage directions and examples for these algorithms are provided in examples directory. To get started with any of the provided algorithms, please follow the notebooks in the the examples directory.

Dependencies

Tested on both Python2.7 and >= Python3.5 with >= Tensorflow 1.6.0.

CPU

pip install -r requirements-cpu.txt

GPU

Install appropriate CUDA and cuDNN [Tested with >= CUDA 8.1 and cuDNN >= 6.1]

pip install -r requirements-gpu.txt

Copyright (c) Microsoft Corporation. All rights reserved. Licensed under the MIT license.