Почему я получаю ошибку о форме при попытке классифицировать изображения с помощью train_generator и fit_generator

Question

При запуске приведенного ниже кода. Я получаю сообщение об ошибке:

ValueError: Error when checking target: expected dense_1 to have shape (1,) but got array with shape (3,).

Может ли кто-нибудь дать мне подсказку.

Код

train_datagen = ImageDataGenerator(rescale=1.0/255)
train_generator = train_datagen.flow_from_directory(
        './marques', 
        target_size=(300, 300),
        color_mode="rgb",
        batch_size=16,
        class_mode='categorical')

model = tf.keras.models.Sequential([
    tf.keras.layers.Conv2D(8, (7,7), activation='relu', input_shape=(300, 300,3)), 
    tf.keras.layers.MaxPooling2D(2, 2),
    tf.keras.layers.Flatten(),
    tf.keras.layers.Dense(128, activation='relu'),
    tf.keras.layers.Dense(3, activation='softmax')
])

model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
model.fit_generator(train_generator, steps_per_epoch=16, epochs=1, verbose=1)