python - 错误的解决方法 while_loop() 得到了一个意外的关键字参数 'maximum_iterations'

Question

当我尝试运行代码时出现以下错误:

model = Sequential()
model.add(LSTM(4, input_shape=(1, look_back)))

TypeError: while_loop() got an unexpected keyword argument 'maximum_iterations'

我知道其中一个解决方案是使用另一个版本的 keras，例如 link .但我使用的是我公司的台式机，每次卸载和安装都必须经过我公司的 IT 部门。是否有另一种解决方法不涉及更改 keras 的版本？

我的keras版本是v2.2.4； tensoflow 是 v1.1.0

Answer 1

有一种方法，但它并不漂亮。您可以深入了解 TensorFlow 内部结构和猴子补丁 while_loop 以忽略 maximum_iterations=:

from tensorflow.python.ops import control_flow_ops

orig_while_loop = control_flow_ops.while_loop

def patched_while_loop(*args, **kwargs):
    kwargs.pop("maximum_iterations", None)  # Ignore.
    return orig_while_loop(*args, **kwargs)


control_flow_ops.while_loop = patched_while_loop

请注意，这不是防弹的，即如果 maximum_iterations 作为位置(而不是关键字)参数给出，它将失败，但它应该“修复”LSTM在你的情况下。