将Keras状态GRU转换为TFLite模型?

时间:2020-07-08 21:12:34

标签: keras tensorflow2.0 tf.keras tensorflow-lite tf-lite

我有一个基本的Keras模型,其中包含stateful=True的GRU层。我想将我的模型转换为TFLite模型,并一次对一个元素的数据进行预测,即,将以大小为1的批处理将一个序列馈送到模型中。查看TensorFlow Docs,没有一种方法可以转换TFLite模型的有状态GRU。但是它确实说(https://www.tensorflow.org/lite/convert/rnn):

仍然可以使用底层无状态Keras LSTM层对有状态Keras LSTM层进行建模,并在用户程序中显式管理状态。使用此处描述的功能,仍然可以将这样的TensorFlow程序转换为TensorFlow Lite。

我不是这个意思。如果我将GRU设置为非有状态,那么如何防止每次预测(批处理)后重置其状态?有什么办法可以防止状态重置?

0 个答案:

没有答案