Machine Learning, Transfer Learning.
model layers and only train the weights of the classification layer. # Freeze the pre-trained model layers for layer in base_model.layers: layer.trainable = False # Compile the model model.compile(optimizer='rmsprop', loss='categorical_crossentropy', metrics=['accuracy']) # Train the model