一直想在手機端訓練模型,一是能夠實作個性化的模型,二是可以避免隐私問題,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