Ошибка при проверке цели: ожидается, что выходные данные будут иметь форму (Нет, 4), но получен массив с формой (30,3)

Я работаю над обучением модели, подобной VGG16, в Керасе. Я использую набор данных TAM и получил эту ошибку. Я читал несколько подобных проблем, но ни один из них не помог мне до сих пор.

Вот код

for layer in model.layers:
    layer.trainable = True

model.compile(optimizer=optimizers.SGD(lr=0.0001, momentum=0.9),
              loss='categorical_crossentropy', metrics = ['accuracy'])

checkpoint = ModelCheckpoint(filepath=model_weights, save_best_only=False, verbose=0)
reduce_lr = ReduceLROnPlateau(monitor='loss', factor=0.2, patience=10, min_lr=0.00001)

history = model.fit_generator(
            generator=train_batches,
            steps_per_epoch=10,
            epochs=5,
            callbacks=[checkpoint, reduce_lr],
            validation_data=valid_batches,
            validation_steps=100,
            verbose=1
)