使用训练好的模型层在keras中创建另一个模型

from tensorflow import keras
from tensorflow.keras import layers

base_model = keras.applications.XXX(weights='imagenet', include_top=False, input_shape=(height, width, channels))

base_model.trainable = False

inputs = keras.Input(shape=(height, width, channels))
x = base_model(inputs, training=False)
x = layers.GlobalAveragePooling2D()(x)
outputs = layers.Dense(num_classes, activation='softmax')(x)
model = keras.Model(inputs, outputs)

model.compile(optimizer='adam', loss='categorical_crossentropy', metrics=['accuracy'])
model.fit(train_dataset, epochs=10, validation_data=val_dataset)