【问题标题】:(tf2/tf.keras) How could # params decreases when a model contains another model(tf2/tf.keras) 当一个模型包含另一个模型时,#params 如何减少?
【发布时间】:2020-02-27 05:54:02
【问题描述】:

我有一个名为 Model A 的 keras 模型(不是层)。模型 A 包含 keras.Layers 类型:Dense、Conv2D、AveragePooling2D、BatchNormalization、add、GlobalAveragePooling2D。

model.summary()的输出如下:

_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
layer_type_1 (Layer_type_1)  multiple                  3776      
_________________________________________________________________
...                           ...                      ...
_________________________________________________________________
dense (Dense)                multiple                  1024      
=================================================================
Total params: 4,787,808
Trainable params: 4,782,496
Non-trainable params: 5,312
_________________________________________________________________

我有另一个 keras 模型(模型 B),它包含模型 A。

总结():

_________________________________________________________________
Layer (type)                 Output Shape              Param #   
=================================================================
conv2d (Conv2D)              multiple                  35        
_________________________________________________________________
model_A (ModelA)             multiple                  4787232   
=================================================================
Total params: 4,787,267
Trainable params: 4,781,955
Non-trainable params: 5,312
_________________________________________________________________

我想知道模型 B 中的总参数数量怎么会少于模型 A?

由于模型 B 包含模型 A,它必须更大。

【问题讨论】:

    标签: python tensorflow keras tf.keras


    【解决方案1】:

    我发现这是因为模型 B 中的第一个 Conv2D 层导致输入形状(模型 A)从 (W, D, 6) 更改为 (W, D, 5)。

    【讨论】:

      猜你喜欢
      • 2013-08-27
      • 2015-12-06
      • 2011-03-04
      • 2016-07-02
      • 2021-04-30
      • 1970-01-01
      • 1970-01-01
      • 2019-02-26
      • 2021-02-20
      相关资源
      最近更新 更多