-
Notifications
You must be signed in to change notification settings - Fork 79
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
#11197: MobilenetV2 bring-up in WH150 with new conv-api
- Loading branch information
1 parent
f3108ee
commit e27db72
Showing
6 changed files
with
978 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,21 @@ | ||
# MobilenetV2 | ||
The MobileNetV2 model is a convolutional neural network (CNN) architecture designed for efficient mobile and embedded vision applications. It was introduced in the paper ["MobileNetV2: Inverted Residuals and Linear Bottlenecks"](https://arxiv.org/abs/1801.04381). </br> | ||
The MobileNetV2 model has been pre-trained on the ImageNet dataset and can be used for various tasks such as image classification, object detection, and semantic segmentation. It has achieved state-of-the-art performance on several benchmarks 1 for mobile and embedded vision applications. | ||
|
||
## How to Run | ||
|
||
To run the demo, make sure to build the project, activate the environment, and set the appropriate environment variables. | ||
For more information, refer [installation and build guide](https://docs.tenstorrent.com/tt-metalium/latest/get_started/get_started.html#install-and-build). | ||
|
||
To run the functional Mobilenetv2 model on a single-chip: | ||
```sh | ||
pytest --disable-warnings models/experimental/functional_mobilenetv2/test/test_ttnn_mobilenetv2.py | ||
``` | ||
|
||
## Supported Hardware | ||
- N150 | ||
|
||
## Other Details | ||
|
||
- Inputs by default are random data. | ||
- The model weights will be automatically downloaded from Google Drive using wget implemented in weights_download.sh. |
Oops, something went wrong.