ValueError: Ошибка при проверке целевой ошибки Keras

Question

Это мой код:

from keras.datasets import mnist
from keras import models
from keras import layers
(train_images,train_lables),(test_images,test_lables) = mnist.load_data()

network = models.Sequential()
network.add(layers.Dense(512,activation='relu',input_shape=(28 * 28,)))
network.add(layers.Dense(10,activation='softmax'))

network.compile(optimizer='rmsprop',
                loss='categorical_crossentropy',
                metrics=['accuracy'])

train_images=train_images.reshape((60000, 28 * 28))
train_images = train_images.astype('float32')/255

test_images = test_images.reshape((10000, 28 * 28))
test_images = test_images.astype('float32')/255

from keras.utils import to_categorical
train_lables = to_categorical(train_lables)
test_lables = to_categorical(test_lables)

network.fit(train_images, train_lables, epochs=5, batch_size=128)

Это дает мне эту ошибку:

ValueError: Ошибка при проверке цели: ожидается, что плотность_4 имеет 2 измерения, но получил массив с формой (60000 , 10, 2)

Что мне делать?