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