ValueError: вход 0 слоя плотности несовместим со слоем: его ранг не определен, но для слоя требуется определенный ранг

При попытке примера на https://www.tensorflow.org/tutorials/images/hub_with_keras на локальном компьютере с tf '2.0.0-alpha0' (скачать, pip-скачать hub '0.4.0-dev' и удалить !pip... строки, установить tf 2, удалить enable_v2_behavior , запустить file), это не удалось с

ValueError: Вход 0 слоя плотности несовместим со слоем: его ранг не определен, но для слоя требуется определенный ранг.

Однако он отлично работает в Google Colab. Почему это и как это можно исправить?