我有一个基本的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设置为非有状态,那么如何防止每次预测(批处理)后重置其状态?有什么办法可以防止状态重置?