首页 > 解决方案 > 在 Tensorflow 2 中循环创建 Keras 模型

问题描述

我使用 Tensorflow 2 在同一代码中有两个 keras 模型。一个模型是在循环外创建的,第二个模型是在循环内创建的。我试图在每次迭代期间使用垃圾收集删除旧模型。但是,在 10-15 次迭代后,代码执行会因内存不足错误而停止。我没有尝试过 tf.keras.backend.clear_session() 因为那也会清除静态模型。当我有另一个静态模型时,在循环中创建 keras 模型的最佳方法是什么?

标签: pythontensorflow2.0tf.keras

解决方案


推荐阅读