Каждый раз умирает kernel (jupyter notebook)

Сразу скажу, что в нейросетях я новичок. У меня есть нейронка (Сделана на keras) которая различает собаку от кошки. Обучающая выборка состоит из 8 тыс. объектов, а тестовоя из 1 тыс. объектов. Все бы ничего но когда я запускаю обучение hist = model.fit(datasetx, datasety, epochs=10, validation_split=0.2), У меня умирает kernel в jupyter notebook. Если что архитектура нейросети: Conv2D(32, (3, 3), padding='same', activation='relu', input_shape=(224, 224, 3)) MaxPooling2D((2, 2), strides=2) Conv2D(64, (3, 3), padding='same', activation='relu') MaxPooling2D((2, 2), strides=2) Dense(128) Dense(1, activation='sigmoid)