Keras错误："BatchNormalization形状必须为1维，但对于batch_normalization却是4维"

Question

3

我有一个Keras函数模型（卷积层神经网络），在使用tensorflow时可以正常运行和拟合。

然而，当使用tensorflow gpu时，该模型无法构建。

以下是构建模型时出现错误的代码行。

x = BatchNormalization(axis=1)(x)

错误信息是 "BatchNormalization Shape必须是秩1，但是对于batch_normalization它却是秩4"。

我认为这是Keras中的一个错误。如果确实如此，那么如何在Windows/Anaconda环境中安装此特定修复程序呢？

Keras版本是2.2.0，tensorflow gpu版本是1.8.0。

有什么想法吗？

提前谢谢大家。

- user898160

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Ioannis Nasios · Accepted Answer

据我所知，您的计算机中存在tensorflow-gpu和keras版本不兼容的问题。

尝试使用keras 2.1.6版本。

pip uninstall keras
pip install -I keras==2.1.6