我正在使用 Keras 进行建模工作,我想知道是否可以通过索引或名称删除某些层?目前我只知道 model.pop()
可以完成这项工作,但它只是删除了最近添加的图层。此外,layers
是 tensorvariable
的类型,我不知道如何删除可以在 numpy array
或 中完成的某些元素>列表
。顺便说一句,我正在使用 Theano 后端。
最佳答案
model.pop() 只是删除最后添加的层是正确的,没有其他记录的删除中间层的方法。
您始终可以像这样获得任何中间层的输出:
base_model = VGG19(weights='imagenet')
model = Model(inputs=base_model.input, outputs=base_model.get_layer('block4_pool').output)
示例取自此处:https://keras.io/applications/
然后在其上添加新图层。
关于python - Keras 在 model.fit() 之后删除图层,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40652929/