加载模型和StopIteration

Question

我正在使用keras R，keras_model_sequential，fit_generator和generator_pred。它运作良好。我保存了模型。但是，当我使用“load_model”并尝试将其用于“generator_pred”时，最终会出现此错误。

  

生成器发生错误：参数'length.out'的长度必须为1   py_call_impl中的错误（可调用，点$ args，点$关键字）：     StopIteration异常：

实际上唯一不同的是我使用重新加载的模型，其他一切都是一样的。有人知道为什么吗？

Answer 1

你可能正在使用“训练”生成器进行“预测”。

• 训练生成器输出X和Y。 （这意味着：长度2）
• 预测生成器仅输出X（表示：长度为1）

如果您知道如何从R中的生成器获取元素，则可以创建包含旧生成器的另一个生成器。新生成器从旧的geneartor获取一个元素，并仅输出第一个元素X，丢弃Y。