在tf会话中运行Keras顺序模型

Question

给定Keras序列模型（特别是2层LSTM）：我们如何在tf会话中运行它？

我必须在一个脚本中多次训练模型，并且我的内存耗尽很快。在tf会话中运行它是正确的解决方案吗？如果不是，那是什么？

Answer 1

在keras中，model.fit（）或model.fit_generator（）通常用于模型训练。可以找到一个示例here。如果您不需要坚持使用tf会话，那么纯keras实现也是一个不错的选择。

由于内存不足问题，可能有多种潜在原因。也许您可以检查数据集是否太大？没有有关您的代码的更多信息，很难说。