Cambie el nombre de la última capa del modelo Keras

model.layers[-1]._name = "layer_name"
FishBrawler