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

привет, я хочу использовать 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)

любое предложение?