Как упомянуто @ сегодня, вы должны сначала проверить этот пост .
conv_3d
: 18464 = 32 * 3 * 3 * 64 (сверточное ядро) +32 (смещение на активацию)
batch_normalization_1
: 128 = 32 * 4
Я считаю, что два параметра в слое нормализации партии являются необучаемыми.Поэтому 64 параметра из bn_1
и 128 параметров из bn_2
- это 192 non-trainable params
в конце.