PyTorch at the Edge: Deploying Over 964 TIMM Models on Android with TorchScript and Flutter

A companion repo for the blog post PyTorch at the Edge: Deploying Over 964 TIMM Models on Android with TorchScript and Flutter.

📂 Folder Structure

• data/ -- Stores all the data to train the model.

• flutter_app/ -- Codes for the Flutter app.

• train/ -- Training notebooks and related files.

⛄ Demo

The following screen recording shows the app running in real time.

Install the pre-built .apk on your Android phone here.

📞 Questions? Connect with me

If you have any questions or feedback, please don't hesitate to reach out to me. I'm active on the following platforms.

❤️ Support Me

I am thrilled to share my work with you and I hope you find it useful.

If you do, please consider supporting my efforts by making a donation and/or sharing this repository on your social media.

Your support will help me to continue developing and maintaining this project, as well as create new ones.

🙏 Acknowledgement

• https://github.com/zezo357/pytorch_lite
• https://github.com/rwightman/pytorch-image-models
• https://www.kaggle.com/code/jhoward/first-steps-road-to-the-top-part-1