deepjavalibrary / djl

An Engine-Agnostic Deep Learning Framework in Java
https://djl.ai
Apache License 2.0
4.07k stars 650 forks source link

Does the framework support Android training? #1075

Open Lizhihzzx opened 3 years ago

Lizhihzzx commented 3 years ago

I'm researching on deep learning frameworks on mobile phone recently.Does this framework support traing on Android os?Thanks.

frankfliu commented 3 years ago

You should be able to use MXNet engine to train your model on Android. Unfortunately, we don't have ready to use Android distribution (.aar) for MXNet on maven. You can follow the instruction to run MXNet engine on Android:

https://github.com/deepjavalibrary/djl-demo/tree/master/android/mxnet-android