如何在Keras中仅打印模型的最后n个层的摘要

时间:2018-10-04 16:42:19

标签: keras

model.summary()打印整个模型的详细信息。有没有办法只打印最后的n层摘要?

如果没有,我可以从现有的预训练模型的后n个层中创建一个新模型,并打印其摘要。

我尝试了以下操作,但可能由于共享输入而出现错误:

temp_model = Model(inputs=base_model.layers[-4].input, outputs = base_model.layers[-1].output)
print(temp_model.summary())

任何帮助将不胜感激。

0 个答案:

没有答案