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