首页 文章

Keras:有没有办法“pop()”顶层?

提问于
浏览
2

在Keras中有一个名为pop()的功能,可以删除模型的底层 . 有没有办法删除模型的顶层?

我有一个完全保存的预先训练的变分自动编码器,我试图只加载解码器(底部四层) .

我正在使用带有Tensorflow后端的Keras .

2 回答

  • 0

    Keras pop()删除最后一个(也就是顶部)图层,而不是底部图层 .

    我建议你使用model.summary()打印出图层列表,然后使用pop()直到只剩下必要的图层 .

  • 1

    pop(0)适合我

    from keras.applications import vgg16
    vgg = vgg16.VGG16(include_top=False, input_shape=(604,604,3))
    vgg.summary()
    vgg.layers.pop(0)
    vgg.summary()
    vgg.layers.pop()
    vgg.summary()
    

相关问题