一直想在手机端训练模型,一是能够实现个性化的模型,二是可以避免隐私问题,tensorflow lite和tensorflow mobile只能run model,不能train model.
之前尝试使用Deeplearning4J, 但是依赖的lib过多,就没有在手机上使用。
前几天发现,居然支持在浏览器里训练模型了。
今天尝试在手机浏览器里run了一下版的mnist训练,在小米6的浏览器里,训练速度非常快。。。。太开心了。
训练demo的link
/tfjs-examples/mnist-core/dist/