Keras错误：“BatchNormalization Shape必须为1级，但batch_normal化为rank 4”

Question

我有一个Keras功能模型（带卷积层的神经网络），它可以很好地处理张量流。 我可以运行它，我可以适应它。

但是，当使用tensorflow gpu时，无法构建模型。

这是构建抛出错误的模型时的代码行。

x = BatchNormalization(axis=1)(x)

并且错误消息是“ BatchNormalization Shape必须为1级，但批次标准化为4级 ”。

在我看来，它是Keras的一个错误。 如果有，如果有修复，我该如何在Windows / Anaconda环境中安装此特定修补程序？

Keras版本是2.2.0和tensorflow gpu 1.8.0

任何的想法？

在此先感谢大家。

Answer 1

在我看来，你的机器中存在tensorflow-gpu-keras版本不兼容。

尝试keras 2.1.6

pip uninstall keras
pip install -I keras==2.1.6