【问题标题】:Is there a way to train a TensorFlow model on iOS?有没有办法在 iOS 上训练 TensorFlow 模型?
【发布时间】:2018-07-05 12:32:06
【问题描述】:

将已训练的 TensorFlow 模型移植到 iOS 的文档定义明确: https://github.com/tensorflow/tensorflow/tree/master/tensorflow/examples/ios

但是,如果模型没有提及:

  • 可以在设备上进行进一步培训,或者
  • 可以从头开始创建并在设备上进行训练

TensorFlow 可以做到这一点吗?

我知道其他提供设备端培训的 Swift/C++ 库,但我对这项技术更感兴趣。

【问题讨论】:

  • @Bhaak 您知道哪些提供设备端培训的“其他 Swift/C++”库?我目前正在寻找不同的库,但找到这样的库真的很难,因为每个人都只是在写关于使用预训练模型的文章......
  • @Dschee 根据网络和技术的类型,您可以使用不同的库。一个例子是 SwiftAI:github.com/Swift-AI/Swift-AI,它只支持完全连接的神经网络。
  • 谢谢@Bghaak,我知道 Swift AI,但正如你所说,它非常有限。它也不再维护(最后一次提交是在 3 年前)。我正在寻找的是基于 CNN 的图像分类器。基本上,我已经使用 Create ML 训练了一些模型,在针对我的用例进行了一分钟的训练后,它提供了足够好的结果。由于 Create ML 在 iOS 上不可用,我需要一个替代方案。我什至会按照很好的说明自己实现它,但我只能在 python 中找到教程,而且很难知道 Swift 中的参数需要是哪种类型...
  • @Dschee 我明白了。对于 CNN 场景,我必须说我都没有找到合适的库。我没有研究过 Apple 本身 (developer.apple.com/documentation/metalperformanceshaders/…) 提供的内核,也不知道限制和目的,但也许这是一个很好的起点?

标签: ios swift tensorflow deep-learning


【解决方案1】:

从 CoreML3 和 UpdatableTask 开始,设备上的训练现在是 API 的一部分:https://developer.apple.com/documentation/coreml/mlupdatetask

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 2019-11-21
    • 2020-01-03
    • 1970-01-01
    • 2018-01-03
    • 2019-01-20
    • 2016-06-02
    • 2019-04-12
    相关资源
    最近更新 更多