Skip to content

Commit 9a40878

Browse files
more fixes
1 parent 2055f75 commit 9a40878

File tree

8 files changed

+24
-20
lines changed

8 files changed

+24
-20
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,9 @@ A tool that adapts models trained by above algorithms to be inferred by fixed po
2929
Applications demonstrating usecases of these algorithms.
3030

3131
### Organization
32-
- The `edgeml_tf` directory contains the specification of these architectures in TensorFlow,
32+
- The `tf` directory contains the `edgeml_tf` package which specifies these architectures in TensorFlow,
3333
and `examples/tf` contains sample training routines for these algorithms.
34-
- The `edgeml_pytorch` directory contains the specification of these architectures in PyTorch,
34+
- The `pytorch` directory contains the `edgeml_pytorch` package which specifies these architectures in PyTorch,
3535
and `examples/pytorch` contains sample training routines for these algorithms.
3636
- The `cpp` directory has training and inference code for Bonsai and ProtoNN algorithms in C++.
3737
- The `applications` directory has code/demonstrations of applications of the EdgeML algorithms.

pytorch/README.md

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,12 +4,12 @@ This directory includes PyTorch implementations of various techniques and
44
algorithms developed as part of EdgeML. Currently, the following algorithms are
55
available in Tensorflow:
66

7-
1. [Bonsai](../docs/publications/Bonsai.pdf)
7+
1. [Bonsai](/docs/publications/Bonsai.pdf)
88
2. S-RNN
9-
3. [FastRNN & FastGRNN](../docs/publications/FastGRNN.pdf)
10-
4. [ProtoNN](../docs/publications/ProtoNN.pdf)
9+
3. [FastRNN & FastGRNN](/docs/publications/FastGRNN.pdf)
10+
4. [ProtoNN](/docs/publications/ProtoNN.pdf)
1111

12-
The PyTorch compute graphs for these algoriths are packaged as `edgeml_pytorch.graph`.
12+
The PyTorch graphs for these algoriths are packaged as `edgeml_pytorch.graph`.
1313
Trainers for these algorithms are in `edgeml_pytorch.trainer`.
1414
Usage directions and examples for these algorithms are provided in
1515
`$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
2020

2121

2222
It is highly recommended that EdgeML be installed in a virtual environment.
23-
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)).
23+
Please create a new virtual environment using your environment manager
24+
([virtualenv](https://virtualenv.pypa.io/en/stable/userguide/#usage) or
25+
[Anaconda](https://docs.conda.io/projects/conda/en/latest/user-guide/tasks/manage-environments.html#creating-an-environment-with-commands)).
2426
Make sure the new environment is active before running the below mentioned commands.
2527

2628
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.
2931
### CPU
3032

3133
```
32-
pip install -r requirements-cpu-pytorch.txt
34+
pip install -r requirements-cpu.txt
3335
pip install -e .
3436
```
3537

@@ -40,7 +42,7 @@ Tested on Python3.6 with >= PyTorch 1.1.0.
4042
Install appropriate CUDA and cuDNN [Tested with >= CUDA 8.1 and cuDNN >= 6.1]
4143

4244
```
43-
pip install -r requirements-gpu-pytorch.txt
45+
pip install -r requirements-gpu.txt
4446
pip install -e .
4547
```
4648

File renamed without changes.
File renamed without changes.

pytorch/setup_python.py renamed to pytorch/setup.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,11 +2,11 @@
22

33
setuptools.setup(
44
name='edgeml',
5-
version='0.2.2',
5+
version='0.3.0',
66
description='PyTorch code for ML algorithms for edge devices developed at Microsoft Research India.',
77
author_email="[email protected]",
8-
packages=['edgeml_pytorch']
8+
packages=['edgeml_pytorch'],
99
license='MIT License',
10-
long_description=open('edgeml_pytorch/README.md').read(),
10+
long_description=open('README.md').read(),
1111
url='https://github.com/Microsoft/EdgeML',
1212
)

tf/README.md

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,13 @@
11
## Edge Machine Learning: Tensorflow Library
22

3-
This directory includes, Tensorflow implementations of various techniques and
3+
This directory includes Tensorflow implementations of various techniques and
44
algorithms developed as part of EdgeML. Currently, the following algorithms are
55
available in Tensorflow:
66

7-
1. [Bonsai](../docs/publications/Bonsai.pdf)
8-
2. [EMI-RNN](../docs/publications/emi-rnn-nips18.pdf)
9-
3. [FastRNN & FastGRNN](../docs/publications/FastGRNN.pdf)
10-
4. [ProtoNN](../docs/publications/ProtoNN.pdf)
7+
1. [Bonsai](/docs/publications/Bonsai.pdf)
8+
2. [EMI-RNN](/docs/publications/emi-rnn-nips18.pdf)
9+
3. [FastRNN & FastGRNN](/docs/publications/FastGRNN.pdf)
10+
4. [ProtoNN](/docs/publications/ProtoNN.pdf)
1111

1212
The TensorFlow compute graphs for these algoriths are packaged as
1313
`edgeml_tf.graph`. Trainers for these algorithms are in `edgeml_tf.trainer`.
@@ -19,7 +19,9 @@ the notebooks in the `examples/tf` directory.
1919
## Installation
2020

2121
It is highly recommended that EdgeML be installed in a virtual environment.
22-
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)).
22+
Please create a new virtual environment using your environment manager
23+
([virtualenv](https://virtualenv.pypa.io/en/stable/userguide/#usage) or
24+
[Anaconda](https://docs.conda.io/projects/conda/en/latest/user-guide/tasks/manage-environments.html#creating-an-environment-with-commands)).
2325
Make sure the new environment is active before running the below mentioned commands.
2426

2527
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.
2830
### CPU
2931

3032
```
31-
pip install -r requirements-cpu-tf.txt
33+
pip install -r requirements-cpu.txt
3234
pip install -e .
3335
```
3436

@@ -39,7 +41,7 @@ Tested on Python3.5 and python 2.7 with >= Tensorflow 1.6.0.
3941
Install appropriate CUDA and cuDNN [Tested with >= CUDA 8.1 and cuDNN >= 6.1]
4042

4143
```
42-
pip install -r requirements-gpu-tf.txt
44+
pip install -r requirements-gpu.txt
4345
pip install -e .
4446
```
4547

File renamed without changes.
File renamed without changes.

0 commit comments

Comments
 (0)