将 Keras 模型转换为 TF Lite

Question

我有一个已保存为 .h5 文件的 Keras 模型。如何将其转换为 TensorFlow Lite 文件？

Model

Error

我一直在教程converter.convert() 中看到这一行 但它只是给了我这个错误 KeyError: 'kernel'

Answer 1

好的，我已经找到了问题所在。问题在于我在模型中使用了 LSTM 层。这是开发人员承认的一个已知错误。

将我的模型更改为仅密集层时，转换效果很好。我希望这可以为人们节省数小时的艰苦研究。