This commit is contained in:
Harsha Vardhan Simhadri 2019-08-22 00:50:56 +05:30
Родитель 2055f758d6
Коммит 9a40878b33
8 изменённых файлов: 24 добавлений и 20 удалений

Просмотреть файл

@ -29,9 +29,9 @@ A tool that adapts models trained by above algorithms to be inferred by fixed po
Applications demonstrating usecases of these algorithms. Applications demonstrating usecases of these algorithms.
### Organization ### Organization
- The `edgeml_tf` directory contains the specification of these architectures in TensorFlow, - The `tf` directory contains the `edgeml_tf` package which specifies these architectures in TensorFlow,
and `examples/tf` contains sample training routines for these algorithms. and `examples/tf` contains sample training routines for these algorithms.
- The `edgeml_pytorch` directory contains the specification of these architectures in PyTorch, - The `pytorch` directory contains the `edgeml_pytorch` package which specifies these architectures in PyTorch,
and `examples/pytorch` contains sample training routines for these algorithms. and `examples/pytorch` contains sample training routines for these algorithms.
- The `cpp` directory has training and inference code for Bonsai and ProtoNN algorithms in C++. - The `cpp` directory has training and inference code for Bonsai and ProtoNN algorithms in C++.
- The `applications` directory has code/demonstrations of applications of the EdgeML algorithms. - The `applications` directory has code/demonstrations of applications of the EdgeML algorithms.

Просмотреть файл

@ -4,12 +4,12 @@ This directory includes PyTorch implementations of various techniques and
algorithms developed as part of EdgeML. Currently, the following algorithms are algorithms developed as part of EdgeML. Currently, the following algorithms are
available in Tensorflow: available in Tensorflow:
1. [Bonsai](../docs/publications/Bonsai.pdf) 1. [Bonsai](/docs/publications/Bonsai.pdf)
2. S-RNN 2. S-RNN
3. [FastRNN & FastGRNN](../docs/publications/FastGRNN.pdf) 3. [FastRNN & FastGRNN](/docs/publications/FastGRNN.pdf)
4. [ProtoNN](../docs/publications/ProtoNN.pdf) 4. [ProtoNN](/docs/publications/ProtoNN.pdf)
The PyTorch compute graphs for these algoriths are packaged as `edgeml_pytorch.graph`. The PyTorch graphs for these algoriths are packaged as `edgeml_pytorch.graph`.
Trainers for these algorithms are in `edgeml_pytorch.trainer`. Trainers for these algorithms are in `edgeml_pytorch.trainer`.
Usage directions and examples for these algorithms are provided in Usage directions and examples for these algorithms are provided in
`$EDGEML_ROOT/examples/pytorch` directory. To get started with any `$EDGEML_ROOT/examples/pytorch` directory. To get started with any
@ -20,7 +20,9 @@ of the provided algorithms, please follow the notebooks in the the
It is highly recommended that EdgeML be installed in a virtual environment. It is highly recommended that EdgeML be installed in a virtual environment.
Please create a new virtual environment using your environment manager ([virtualenv](https://virtualenv.pypa.io/en/stable/userguide/#usage) or [Anaconda](https://docs.conda.io/projects/conda/en/latest/user-guide/tasks/manage-environments.html#creating-an-environment-with-commands)). Please create a new virtual environment using your environment manager
([virtualenv](https://virtualenv.pypa.io/en/stable/userguide/#usage) or
[Anaconda](https://docs.conda.io/projects/conda/en/latest/user-guide/tasks/manage-environments.html#creating-an-environment-with-commands)).
Make sure the new environment is active before running the below mentioned commands. Make sure the new environment is active before running the below mentioned commands.
Use pip to install requirements before installing the `edgeml_pytorch` library. Use pip to install requirements before installing the `edgeml_pytorch` library.
@ -29,7 +31,7 @@ Details for cpu based installation and gpu based installation provided below.
### CPU ### CPU
``` ```
pip install -r requirements-cpu-pytorch.txt pip install -r requirements-cpu.txt
pip install -e . pip install -e .
``` ```
@ -40,7 +42,7 @@ Tested on Python3.6 with >= PyTorch 1.1.0.
Install appropriate CUDA and cuDNN [Tested with >= CUDA 8.1 and cuDNN >= 6.1] Install appropriate CUDA and cuDNN [Tested with >= CUDA 8.1 and cuDNN >= 6.1]
``` ```
pip install -r requirements-gpu-pytorch.txt pip install -r requirements-gpu.txt
pip install -e . pip install -e .
``` ```

Просмотреть файл

Просмотреть файл

Просмотреть файл

@ -2,11 +2,11 @@ import setuptools #enables develop
setuptools.setup( setuptools.setup(
name='edgeml', name='edgeml',
version='0.2.2', version='0.3.0',
description='PyTorch code for ML algorithms for edge devices developed at Microsoft Research India.', description='PyTorch code for ML algorithms for edge devices developed at Microsoft Research India.',
author_email="edgeml@microsoft.com", author_email="edgeml@microsoft.com",
packages=['edgeml_pytorch'] packages=['edgeml_pytorch'],
license='MIT License', license='MIT License',
long_description=open('edgeml_pytorch/README.md').read(), long_description=open('README.md').read(),
url='https://github.com/Microsoft/EdgeML', url='https://github.com/Microsoft/EdgeML',
) )

Просмотреть файл

@ -1,13 +1,13 @@
## Edge Machine Learning: Tensorflow Library ## Edge Machine Learning: Tensorflow Library
This directory includes, Tensorflow implementations of various techniques and This directory includes Tensorflow implementations of various techniques and
algorithms developed as part of EdgeML. Currently, the following algorithms are algorithms developed as part of EdgeML. Currently, the following algorithms are
available in Tensorflow: available in Tensorflow:
1. [Bonsai](../docs/publications/Bonsai.pdf) 1. [Bonsai](/docs/publications/Bonsai.pdf)
2. [EMI-RNN](../docs/publications/emi-rnn-nips18.pdf) 2. [EMI-RNN](/docs/publications/emi-rnn-nips18.pdf)
3. [FastRNN & FastGRNN](../docs/publications/FastGRNN.pdf) 3. [FastRNN & FastGRNN](/docs/publications/FastGRNN.pdf)
4. [ProtoNN](../docs/publications/ProtoNN.pdf) 4. [ProtoNN](/docs/publications/ProtoNN.pdf)
The TensorFlow compute graphs for these algoriths are packaged as The TensorFlow compute graphs for these algoriths are packaged as
`edgeml_tf.graph`. Trainers for these algorithms are in `edgeml_tf.trainer`. `edgeml_tf.graph`. Trainers for these algorithms are in `edgeml_tf.trainer`.
@ -19,7 +19,9 @@ the notebooks in the `examples/tf` directory.
## Installation ## Installation
It is highly recommended that EdgeML be installed in a virtual environment. It is highly recommended that EdgeML be installed in a virtual environment.
Please create a new virtual environment using your environment manager ([virtualenv](https://virtualenv.pypa.io/en/stable/userguide/#usage) or [Anaconda](https://docs.conda.io/projects/conda/en/latest/user-guide/tasks/manage-environments.html#creating-an-environment-with-commands)). Please create a new virtual environment using your environment manager
([virtualenv](https://virtualenv.pypa.io/en/stable/userguide/#usage) or
[Anaconda](https://docs.conda.io/projects/conda/en/latest/user-guide/tasks/manage-environments.html#creating-an-environment-with-commands)).
Make sure the new environment is active before running the below mentioned commands. Make sure the new environment is active before running the below mentioned commands.
Use pip to install the requirements before installing the `edgeml_tf` library. Use pip to install the requirements before installing the `edgeml_tf` library.
@ -28,7 +30,7 @@ Details for cpu based installation and gpu based installation provided below.
### CPU ### CPU
``` ```
pip install -r requirements-cpu-tf.txt pip install -r requirements-cpu.txt
pip install -e . pip install -e .
``` ```
@ -39,7 +41,7 @@ Tested on Python3.5 and python 2.7 with >= Tensorflow 1.6.0.
Install appropriate CUDA and cuDNN [Tested with >= CUDA 8.1 and cuDNN >= 6.1] Install appropriate CUDA and cuDNN [Tested with >= CUDA 8.1 and cuDNN >= 6.1]
``` ```
pip install -r requirements-gpu-tf.txt pip install -r requirements-gpu.txt
pip install -e . pip install -e .
``` ```

Просмотреть файл

Просмотреть файл