Skip to content

Commit

Permalink
#158 Change jupyter notebook to google colab (#160)
Browse files Browse the repository at this point in the history
  • Loading branch information
Only-bottle authored Feb 3, 2024
1 parent 59dd82e commit c53a09f
Show file tree
Hide file tree
Showing 4 changed files with 12 additions and 13 deletions.
4 changes: 4 additions & 0 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -1,5 +1,8 @@
FROM python:3.8.16

ARG TENSORFLOW_VERSION="2.8.0"
ARG PROTOBUF_VERSION="3.20.2"

RUN apt-get update && \
apt-get install -y \
git \
Expand All @@ -17,6 +20,7 @@ RUN apt-get update && \
apt-get clean && apt-get autoremove && rm -rf /var/lib/apt/lists/*

RUN python -m pip install --upgrade pip
RUN python -m pip install --no-cache-dir tensorflow-gpu==${TENSORFLOW_VERSION} protobuf==${PROTOBUF_VERSION} && rm -rf /root/.cache/pip

RUN mkdir -p /home/appuser/netspresso
WORKDIR /home/appuser/netspresso
Expand Down
10 changes: 3 additions & 7 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -32,8 +32,7 @@
<a href="https://www.tensorflow.org/install/pip" target="_blank"><img src="https://img.shields.io/badge/TensorFlow-2.3.x ~ 2.8.x.-FF6F00?style=flat&logo=tensorflow&logoColor=#FF6F00&link=https://www.tensorflow.org/install/pip"/></a>
<br>
<a href="https://netspresso.ai?utm_source=git&utm_medium=text_np&utm_campaign=py_launch"><img src="https://img.shields.io/badge/NetsPresso-Open in Website-1BD2EB?style=flat&link=https://netspresso.ai/"/></a>
<a href="https://github.com/Nota-NetsPresso/NetsPresso-Model-Compressor-ModelZoo"><img src="https://img.shields.io/badge/ModelZoo-Open in Github-181717?style=flat&logo=github&logoColor=#181717"/></a>
<a href="https://github.com/Nota-NetsPresso/NetsPresso-Model-Compressor-ModelZoo/tree/main/best_practices"><img src="https://img.shields.io/badge/Best Practice-Open in Colab-F9AB00?style=flat&logo=googlecolab&logoColor=#F9AB00"/></a>
<a href="https://colab.research.google.com/drive/15HBp88qfUDQl5PaEcZ5J-dnLo-1kvw-a#scrollTo=-SO6lDi5-WY0" target="_blank"><img src="https://img.shields.io/badge/Best Practice-Open in Colab-F9AB00?style=flat&logo=googlecolab&logoColor=#F9AB00"/></a>
</p>
</div>
</br>
Expand All @@ -44,6 +43,8 @@ NetsPresso resolves AI-related constraints in business use cases and enables cos
Easily compress various models with our resources. Please browse the [Docs] for details, and join our [Discussion Forum] for providing feedback or sharing your use cases.

To get started with the NetsPresso, you will need to sign up either at [NetsPresso].</a>

We offer a comprehensive guide to walk you through the process of optimizing an AI model using NetPresso. A full tutorial can be found [Google Colab](https://colab.research.google.com/drive/15HBp88qfUDQl5PaEcZ5J-dnLo-1kvw-a#scrollTo=-SO6lDi5-WY0).
</br>
</br>

Expand Down Expand Up @@ -454,10 +455,6 @@ print(f"model cpu memory footprint: {benchmark_result['result']['memory_footprin
</tr>
</table>

## NetsPresso Model Compressor Best Practice

If you want to experience Model Compressor online without any installation, please refer to the [NetsPresso-Model-Compressor-ModelZoo] repo that runs on Google Colab.

## Contact

Join our [Discussion Forum] for providing feedback or sharing your use cases, and if you want to talk more with Nota, please contact us [here].</br>
Expand All @@ -470,4 +467,3 @@ Or you can also do it via email([[email protected]]) or phone(+82 2-555-8659)!
[NetsPresso-Sign-Up]: https://netspresso.ai/signup
[here]: https://www.nota.ai/contact-us
[netspresso@nota.ai]: mailto:netspresso@nota.ai
[NetsPresso-Model-Compressor-ModelZoo]: https://github.com/Nota-NetsPresso/NetsPresso-Model-Compressor-ModelZoo
5 changes: 3 additions & 2 deletions docker-compose.yml
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,9 @@ services:
ports:
# HOST_PORT for tensorboard
- "50001:50001" # tensorboard
- "50002:50002" # (optional, gradio) configuration helper
- "50003:50003" # (optional, gradio) inference demo
- "50002:50002" # jupyter notebook
- "50003:50003" # (optional, gradio) configuration helper
- "50004:50004" # (optional, gradio) inference demo
volumes:
# from path: your working directory
# - /PATH/TO/netspresso:/home/appuser/netspresso
Expand Down
6 changes: 2 additions & 4 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -2,12 +2,10 @@ loguru==0.7.0
urllib3==2.0.2
PyJWT==2.7.0
pydantic==1.10.4
requests==2.30.0
requests>=2.30.0
email-validator==2.0.0
pytz==2023.3
pytz>=2023.3
typing_extensions==4.5.0
netspresso_trainer==0.1.2
PyGithub>=2.1.1
matplotlib>=3.7.4
tensorflow-gpu==2.8
protobuf<=3.20.2

0 comments on commit c53a09f

Please sign in to comment.