Хорошо ли подсчитывает параметры функция keras.Sequential.summary ()? - PullRequest
0 голосов
/ 12 июня 2019

Я использую функцию summary (), чтобы получить профиль сети и получить общие параметры, и я обнаружил, что это ошибка.

Я не использовал никаких функций печати, и способ вызова функции выглядит следующим образом: "model.summary (line_length = 36, позиции = [15, 43, 40])"

Чистая A:

    ____________________________________
    Layer (type)   Output Shape             
    ====================================
    conv2d_170 (Co (None, 28, 28, 128)      
    ____________________________________
    max_pooling2d_ (None, 14, 14, 128)      
    ____________________________________
    flatten_68 (Fl (None, 25088)            
    ____________________________________
    dense_132 (Den (None, 128)              
    ____________________________________
    dense_133 (Den (None, 128)              
    ____________________________________
    dropout_68 (Dr (None, 128)              
    ____________________________________
    dense_134 (Den (None, 10)               
    ====================================
    Total params: 3,232,522
    Trainable params: 3,232,522
    Non-trainable params: 0

Нетто B:

    ____________________________________
    Layer (type)   Output Shape             
    ====================================
    conv2d_176 (Co (None, 28, 28, 128)      
    ____________________________________
    max_pooling2d_ (None, 14, 14, 128)      
    ____________________________________
    conv2d_177 (Co (None, 13, 13, 128)      
    ____________________________________
    max_pooling2d_ (None, 7, 7, 128)        
    ____________________________________
    conv2d_178 (Co (None, 6, 6, 128)        
    ____________________________________
    max_pooling2d_ (None, 3, 3, 128)        
    ____________________________________
    conv2d_179 (Co (None, 2, 2, 128)        
    ____________________________________
    max_pooling2d_ (None, 1, 1, 128)        
    ____________________________________
    flatten_71 (Fl (None, 128)              
    ____________________________________
    dense_141 (Den (None, 128)              
    ____________________________________
    dense_142 (Den (None, 128)              
    ____________________________________
    dropout_71 (Dr (None, 128)              
    ____________________________________
    dense_143 (Den (None, 10)               
    ====================================
    Total params: 234,634
    Trainable params: 234,634
    Non-trainable params: 0

Можно ожидать, что сеть B будет иметь больше обучаемых параметров, чего не происходит.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...