Ошибка Keras: «Форма BatchNormalization должна иметь ранг 1, но имеет ранг 4 для batch_normalization»

Question

У меня есть функциональная модель Keras (нейронная сеть со сверточными слоями), которая отлично работает с тензорным потоком.Я могу запустить это, и я могу соответствовать этому.

Тем не менее, модель не может быть построена при использовании Тензор потока GPU.

Это строка кода при построении модели, которая выдает ошибку.

x = BatchNormalization(axis=1)(x)

И сообщение об ошибке: « BatchNormalization Shape должно иметь ранг 1, но является рангом 4 для batch_normalization ».

Мне кажется, что это ошибка в Keras.Если да, то если есть исправление, как я могу установить это исправление в среде Windows / Anaconda?

Версия Keras - 2.2.0, а tenorflow gpu 1.8.0

Есть идеи?

Спасибо всем заранее.

Answer 1

Мне кажется, что на вашей машине есть несовместимость с tenorsflow-gpu - keras.

Попробуйте керас 2.1.6

pip uninstall keras
pip install -I keras==2.1.6