2017-06-18 79 views
1

在Keras中有一个名为pop()的功能,可以让您删除模型的底层。有什么办法去除模型的顶层?Keras:有没有办法“弹出()”顶层?

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

我使用Keras和Tensorflow后端。

+1

你试过'流行(0)'? –

回答

0

Keras pop()删除最后一层(又名顶层),而不是底层。

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

相关问题