Skip to content

Commit

Permalink
#11197: MobilenetV2 bring-up in WH150 with new conv-api
Browse files Browse the repository at this point in the history
  • Loading branch information
keerthana-r-mcw committed Nov 26, 2024
1 parent f3108ee commit e27db72
Show file tree
Hide file tree
Showing 6 changed files with 978 additions and 0 deletions.
21 changes: 21 additions & 0 deletions models/experimental/functional_mobilenetv2/README.md
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.
Loading

0 comments on commit e27db72

Please sign in to comment.