модуль tenensflow_hub не имеет атрибута KerasLayer - PullRequest
Новая
       11

модуль tenensflow_hub не имеет атрибута KerasLayer

0 голосов
/ 23 апреля 2019

Когда я пытаюсь переобучить модель с помощью tenorflow, она выдает ошибку: 

**error module 'tensorflow_hub' has no attribute 'KerasLayer'**

Код: 

print("Building model with", MODULE_HANDLE)
model = tf.keras.Sequential([
    hub.KerasLayer(MODULE_HANDLE, output_shape=[FV_SIZE],
    trainable=do_fine_tuning),
    tf.keras.layers.Dropout(rate=0.2),
    tf.keras.layers.Dense(train_generator.num_classes,
    activation='softmax',
    kernel_regularizer=tf.keras.regularizers.l2(0.0001))
])
model.build((None,)+IMAGE_SIZE+(3,))
model.summary()

Ошибка как: 

      1 print("Building model with", MODULE_HANDLE)
      2 model = tf.keras.Sequential([
----> 3     hub.KerasLayer(MODULE_HANDLE, output_shape=[FV_SIZE],
      4                    trainable=do_fine_tuning),
      5     tf.keras.layers.Dropout(rate=0.2),

AttributeError: module 'tensorflow_hub' has no attribute 'KerasLayer'

с помощью концентратора тензорного потока переобучите предыдущую модель концентратора, добавив новые полностью подключенные слои dence. Когда запускаете код, он показывает вышеуказанную ошибку. Любое есть представление об этом. Пожалуйста, помогите

1 Ответ

0 голосов
/ 23 апреля 2019

Пожалуйста, проверьте версию tenorflow. Это должна быть последняя ночная версия.

Когда я использую версию, подобную 1.13.1, я вижу следующее предупреждение перед ошибкой, без атрибута 'KerasLayer': 

W0423 20:04:16.453974 139707130586880 __init__.py:56] Some hub symbols are not available because TensorFlow version is less than 1.14

После выполнения pip install "tf-nightly" все отлично работает.

https://www.tensorflow.org/hub

Для проблемы BatchNormalizationv1 вы можете использовать tf2.0 по ночам, которая также должна позаботиться об исходной проблеме. 

pip install -U tf-nightly-2.0-preview

https://github.com/tensorflow/tfjs/issues/1255

...