手机上训练机器学习模型

时间:2025-01-30 07:06:19

一直想在手机端训练模型,一是能够实现个性化的模型,二是可以避免隐私问题,tensorflow lite和tensorflow mobile只能run model,不能train model.

之前尝试使用Deeplearning4J, 但是依赖的lib过多,就没有在手机上使用。

前几天发现,居然支持在浏览器里训练模型了。

今天尝试在手机浏览器里run了一下版的mnist训练,在小米6的浏览器里,训练速度非常快。。。。太开心了。

训练demo的link

/tfjs-examples/mnist-core/dist/