Как использовать efficien tnet -lite, предоставленный tfhub для второго тренинга на tf2.1

Question

Я использую версию tennsflow-gpu 2.1.0, установленную из pip. '' 'импортировать тензорный поток как tf импортировать тензорный поток как hub

tf.keras.backend.set_learning_phase(True)

module_url = "https://tfhub.dev/tensorflow/efficientnet/lite0/classification/2"

module2 = tf.keras.Sequential([
    hub.KerasLayer(module_url, trainable=False, input_shape=(224,224,3))])

output1 = module2(tf.ones(shape=(1,224,224,3)))
print(module2.summary())

' ''

Когда я установил trainable = True, операция выдаст ошибку. Итак, я не могу переучить его на версии tf2.1?