Как исправить ValueError: при добавлении tensorflow_hub.KerasLayer в последовательную модель можно добавить только экзем ⇐ Python

[Anonymous]

Я изучаю TensorFlow и трансферное обучение и пытаюсь добавить экстрактор функций TensorFlow Hub в модель Keras Sequential. Но я получаю эту ошибку:

ValueError: в последовательную модель можно добавить только экземпляры keras.Layer.

Мой код:

Код: Выделить всё

import tensorflow as tf
import tensorflow_hub as hub
from tensorflow.keras import layers

def create_model(model_url, num_classes=10):
model = tf.keras.Sequential([
hub.KerasLayer(model_url, trainable=False, input_shape=(224,224,3)),
layers.Dense(num_classes, activation="softmax")
])
return model

resnet_model = create_model(resnet_url,
num_classes=train_data_10_percent.num_classes)

Я ожидал, что Hub.KerasLayer будет работать как обычный слой Keras, но Sequential выдает эту ошибку.
Почему это происходит и как правильно использовать модели TF Hub внутри модели Sequential?>

Подробнее здесь: https://stackoverflow.com/questions/798 ... -sequentia