繁体   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