恢复没有最后一层的tensorflow Resnet50V2模型答案

【问题标题】：Restoring tensorflow Resnet50V2 model without last layer恢复没有最后一层的tensorflow Resnet50V2模型
【发布时间】：2020-07-18 15:02:58
【问题描述】：

我想从检查点恢复 Resnet50V2 模型，但不希望在新模型中恢复最后一层。我不明白该怎么做？伪代码如下：

我将第一个模型训练为

  Model= ResNet50v2 ()

 Cnn_model= Sequential([
                        Average pooling layer,
                        Dense (2048),
                        Dense (2048).
                        Dense (10) ])

现在我想使用相同架构的上述模型的检查点，但不想添加 Dense(10) 层。

【问题讨论】：

这是一个重复的问题，请参阅此处stackoverflow.com/questions/55392591/…
感谢您的回复。我以前见过这个问题。我正在编辑我的问题，以便更清楚。

标签： tensorflow

【解决方案1】：

您可以使用 Sequential 类中的 layers 方法实例化一个新模型，并在从检查点加载后获取所有层，直到最后一层：

model = Sequential([i for i in Cnn_model.layers[:-1])

【讨论】：

感谢您的回答。有没有办法直接将权重恢复到新模型（不包括最后一层）而不是先恢复到预训练模型。