gpt4 book ai didi

python - 如何清除使用 Keras 和 Tensorflow(作为后端)创建的模型?

转载 作者:太空狗 更新时间:2023-10-29 21:21:42 24 4
gpt4 key购买 nike

我在 Jupyter Notebook 中使用 Keras 训练神经网络时遇到问题。我创建了一个具有多个隐藏层的顺序模型。训练模型并保存结果后,我想删除该模型并在同一 session 中创建一个新模型,因为我有一个 for 循环来检查不同参数的结果。但是据我了解我得到的错误,当我改变参数时,当我循环时,我只是向模型添加层(即使我在循环内用 network = Sequential() 再次初始化它).所以我的问题是,我怎样才能完全清除以前的模型,或者我怎样才能在同一个 session 中初始化一个全新的模型?

最佳答案

keras.backend.clear_session() 应该清除之前的模型。来自 https://keras.io/backend/ :

Destroys the current TF graph and creates a new one. Useful to avoid clutter from old models / layers.

关于python - 如何清除使用 Keras 和 Tensorflow(作为后端)创建的模型?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52133347/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com