2017-06-18 92 views
1

在Keras中有一個名爲pop()的功能,可以讓您刪除模型的底層。有什麼辦法去除模型的頂層?Keras:有沒有辦法「彈出()」頂層?

我有一個完全保存的預訓練變分自動編碼器,並試圖只加載解碼器(底部四層)。

我使用Keras和Tensorflow後端。

+1

你試過'流行(0)'? –

回答

0

Keras pop()刪除最後一層(又名頂層),而不是底層。

我建議你使用model.summary()來打印出圖層列表,然後使用pop()直到只剩下必要的圖層。

相關問題