2 Likes
Just provide an easy way to export trained models in pure PyTorch format.
Then we can directly use Pytorch mobile.
2 Likes
This should already be doable. learn.save('model.pth')
will save the model state_dict.
2 Likes
image classifiers can be exported easily, just follow the steps from the Pytorch mobile website. I am having trouble exporting detectors like Retinanet.
Have a look at the docs of https://course.fast.ai/deployment_seeme_ai deploying your fastai model on the web and mobile.
TLDR: Once uploaded, your model is automatically converted to the different formats and you use your model on iOS and Android apps without any further development.
Disclaimer: Iām the creator of SeeMe.ai, but very happy to assist or answer questions.