python - 如何使 MLModel 在 Swift CoreML 中可更新？

Question

我正在尝试将可更新模型的新 CoreML 3 功能实现到我的应用程序中，但我无法弄清楚。

我在 Keras 中创建了一个具有 2 层的神经网络，并将其转换为核心 ML 模型。然后我将该模型包含到我的 iOS 项目中。然而 self.testmodel.model.modelDescription.isUpdatable 总是错误的，我无法理解苹果文档。

model = Sequential([
    Dense(10, activation="sigmoid", input_shape=(2,)),
    Dense(2, activation="relu"),
])
core_mlmodel = coremltools.converters.keras.convert(model)
core_mlmodel.save("FirstNN.mlmodel")

集成在 Swift 中，我现在可以使用模型，但无法更新它

let testmodel = FirstNN()
try testmodel.prediction(input: input) // works
testmodel.model.modelDescription.isUpdatable  // is false

为什么我的模型不可更新，我该如何更改？

Answer 1

您可以将 respect_trainable=True 参数传递给 coremltools.converters.keras.convert()。

或者，您可以在转换后更改 mlmodel 文件以使模型可更新。示例形式的官方文档在这里:https://github.com/apple/coremltools/tree/master/examples/updatable_models