Keras清除所有gpu内存

Question

我在做这样的事情：

for ai in ai_generator:
   ai.fit(ecc...)

ai_generator是一个生成器，用于实例化具有不同配置的模型。

我的问题是gpu内存溢出，K.clear_session()不能工作，因为它会抛出

ValueError: Tensor("conv2d_1/kernel:0", shape=(3, 3, 1, 1), dtype=float32_ref) must be from the same graph as Tensor("concat:0", shape=(?, 38, 300, 1), dtype=float32).

我怎样才能把角星的记忆清除为新的记忆？我只需要性能适合的方法，我可以删除所有。

Answer 1

我解决了删除模型之间共享的所有层的问题。“共享”实例是输入。然后我做了这个：

for ai in aigen:
   ai.fit(**params)
   del ai #  for avoid any trace on aigen
   tf.reset_default_graph() # for being sure
   K.clear_session() # removing session, it will instance another