Керас: тренировка застряла из-за потери категорийной кросентропии

Question

Я хочу обучить модель по керасу, и сеть:

    model = DenseNet121(
        include_top=True,
        weights=None,
        input_shape=[shape[0], shape[1], 3],
        classes=4,
    )
    model.summary()
    optimizer = keras.optimizers.Adam(learning_rate=0.0001)
    model.compile(
            optimizer=optimizer,
            loss=keras.losses.categorical_crossentropy,
            metrics=['accuracy']
        )

Я уверен, что в коде нет грамматической ошибки. На тренировках он застрял в 3-й эпохе на 180 партий. Тренировка просто останавливается без ошибок.

Более странно то, что тренировка в порядке, когда я использую binary_crossentropy в качестве потери. Я действительно в замешательстве. Как я могу исправить эту ошибку? Любое предложение приветствуется.

Среда:

win10
tensorflow 1.14.0
keras 2.3.1