dense layer берем 1024 units, units - это размерность выходного пространства, как написано в доках. Поч именно 1024, мне не понятно, какая-то формула есть?
У меня на входе 28 на 28 фотка, после 2 пулов осталось 7*7*64, что получается 3136 фич
Они просто делают преобразование данных из фильтра с дропаутом. 1024 взято скорее всего с помощью перебора оптимальных параметров. Дропаут чтобы сеточка не переобучалась. А сам выходной слой у тебя состоит из 10 нейронов, каждый из которых даёт вероятность принадлежности input к одному из классов.
из 3136 пережимается в вектор 1024, далее из 1024 в 10
Обсуждают сегодня