Skip to content

Update README.md #8

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
avishekchy45 wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Update README.md #8

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ pip install model-profiler
or latest version from [PyPI project site](https://pypi.org/project/model-profiler/)
## Usage

Firs load any model built using keras or tensorflow. Here for simplicity we will load model from kera applications.
First load any model built using keras or tensorflow. Here for simplicity we will load model from keras applications.

```python
form tensorflow.keras.applications import VGG16
Expand Down Expand Up @@ -69,20 +69,20 @@ print(profile)
| Model Parameters | 138.357544 | Million |
| Memory Required by Model Weights | 527.7921447753906 | MB |
```
Default units for the prfiler are
Default units for the profiler are

```
# in order
use_units = ['GPU IDs', 'BFLOPs', 'GB', 'Million', 'MB']

```
You can change units by changing the list entry in appropriate location. For example if you want to get `model` FLOPs in million just change the list as follows.
You can change units by changing the list entry in appropriate location. For example, if you want to get `model` FLOPs in millions just change the list as follows.

```
# keep order
use_units = ['GPU IDs', 'MFLOPs', 'GB', 'Million', 'MB']
```
### Availabel units are
### Available units are
```
'GB':memory unit gega-byte
'MB': memory unit mega-byte
Expand Down