在Keras中有一个名为pop()的功能,可以删除模型的底层 . 有没有办法删除模型的顶层?
我有一个完全保存的预先训练的变分自动编码器,我试图只加载解码器(底部四层) .
我正在使用带有Tensorflow后端的Keras .
Keras pop()删除最后一个(也就是顶部)图层,而不是底部图层 .
我建议你使用model.summary()打印出图层列表,然后使用pop()直到只剩下必要的图层 .
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()
2 回答
Keras pop()删除最后一个(也就是顶部)图层,而不是底部图层 .
我建议你使用model.summary()打印出图层列表,然后使用pop()直到只剩下必要的图层 .
pop(0)适合我