标签: keras
model.summary()打印整个模型的详细信息。有没有办法只打印最后的n层摘要?
model.summary()
如果没有,我可以从现有的预训练模型的后n个层中创建一个新模型,并打印其摘要。
我尝试了以下操作,但可能由于共享输入而出现错误:
temp_model = Model(inputs=base_model.layers[-4].input, outputs = base_model.layers[-1].output) print(temp_model.summary())
任何帮助将不胜感激。