簡體   English   中英

如何清除使用 Keras 和 Tensorflow(作為后端)創建的 model?

[英]How can I clear a model created with Keras and Tensorflow(as backend)?

我在 Jupyter Notebook 中使用 Keras 訓練 neural.net 時遇到問題。 我創建了一個帶有多個隱藏層的順序 model。 在訓練 model 並保存結果后,我想刪除這個 model 並在同一個 session 中創建一個新的 model,因為我有一個for循環來檢查不同參數的結果。 但是據我了解我得到的錯誤,當更改參數時,當我循環時,我只是向 model 添加層(即使我在循環內使用network = Sequential()再次初始化它)。 所以我的問題是,我怎樣才能完全清除以前的 model 或者我怎樣才能在同一個 session 中初始化一個全新的 model?

keras.backend.clear_session()應該清除先前的模型。 https://keras.io/backend/

銷毀當前的TF圖並創建一個新的圖。 有用以避免舊模型/圖層混亂。

我知道這是一個有點舊的線程,但我正在尋找清除 session 的東西。對於TensorFlow 2.8 ,我認為你需要使用tf.keras.backend.clear_session()

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM