На моем dell core i7 - 16 ГБ ОЗУ - 4 ГБ 960-метровый ноутбук с графическим процессором я работаю над проектом по классификации изображений КТ легких с использованием 3D CNN. Я использую процессорную версию tenorflow. Изображения подготовлены как размер массива (25,50,50).
В моей модели CNN было 2 слоя конвона, два слоя maxpool, один слой FC и выходной слой. С этой архитектурой я мог обучить модель приблизительно (от 5000 до 6000) образцов. После добавления дополнительных слоев моя модель теперь имеет 6 слоев конвоя, 3 слоя максимального пула, FC и выходной слой. Моя проблема в том, что после изменения архитектуры с более чем 1000 выборок моя память заполняется, и я получаю ошибку памяти. Я пытался делать меньшие партии, но каждый раз получал одну и ту же ошибку. У меня есть два вопроса:
Почему при добавлении большего количества слоев модели требуется больше памяти?
Есть ли способ решить эту проблему?