Ошибка автоматического перезапуска во время выполнения в Google Colab

Question

Привет, я хочу использовать Google Colab для обучения модели для моей классификации, но когда я начинаю тренировку через несколько секунд, он говорит: «Время выполнения умерло. Автоматический перезапуск. '

моя модель:

model=keras.Sequential([
    keras.layers.Flatten(input_shape=(230, 320, 3)),
    keras.layers.Dense(32,activation=tf.nn.relu),
    keras.layers.Dense(10,activation=tf.nn.relu),
    keras.layers.Dense(5,activation=tf.nn.softmax),
])
opt=tf.train.AdamOptimizer()
model.compile(loss='sparse_categorical_crossentropy',optimizer=opt,metrics=['accuracy'])

и обучающий код:

model.fit(train_data,train_labels,epochs=25,validation_split=0.2)

есть предложения?

Answer 1

, поэтому я решил проблему, и ошибка была в том, что я выбрал один канал изображения, когда я выбрал 3 канала, проблема исчезла.