[英]Keras Functional API changing layer names in every API
這是一個非常好的問題,它說明了有關keras
一些真正重要的問題。 名稱以這種方式更改的原因是,即使您覆蓋模型, keras
也不會清除先前定義的變量。 您可以通過調用以下命令輕松檢查變量是否仍在session.graph
:
from keras import backend as K
K.get_session().graph.get_collection('variables')
為了清除先前的模型變量,可以調用:
K.clear_session()
但是-要小心-因為您可能會丟失現有模型。 如果要保持名稱不變,只需在圖層實例中添加name
參數即可為圖層name
,例如:
Dense(10, activation='softmax', name='output')
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.