Я реализовал пользовательский CNN с комбинацией VGG16 Conv и пользовательского FC для задачи классификатора изображений.Вывод моего классификатора 2 класса.
Я использовал 2 плотных слоя, а архитектура плотных слоев описана ниже.
Ввод 25088 => Dense1 (1024 узла) => Dense2 (2 узла)
СУ этой архитектуры у меня размер модели ~ 300МБ.Я не мог запустить огромную модель из-за ограничения памяти.Поэтому я уменьшил Dense1 1024 узла до 128 узлов.
Здесь у меня есть 2 вопроса,
- Это правильный подход для уменьшения количества узлов в моем решении?
- Как выбрать число узлов в плотнойслой?
Заранее спасибо.