天天看點

手機上訓練機器學習模型

一直想在手機端訓練模型,一是能夠實作個性化的模型,二是可以避免隐私問題,tensorflow lite和tensorflow mobile隻能run model,不能train model.

之前嘗試使用Deeplearning4J, 但是依賴的lib過多,就沒有在手機上使用。

前幾天發現,tensorflow.js居然支援在浏覽器裡訓練模型了。

今天嘗試在手機浏覽器裡run了一下tensorflow.js版的mnist訓練,在小米6的浏覽器裡,訓練速度非常快。。。。太開心了。

訓練demo的link

https://storage.googleapis.com/tfjs-examples/mnist-core/dist/index.html

繼續閱讀