将Keras模型转换为TF Lite

时间:2019-11-20 07:50:44

标签: tensorflow machine-learning keras tensorflow2.0 tensorflow-lite

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

Model

Error

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

1 个答案:

答案 0 :(得分:0)

好的,我已经解决了问题。问题在于我在模型中使用LSTM层。这是开发人员认可的已知错误Here

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