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).
如何将keras内存清除为新内容？ 我只需要适合方法的表现，我可以删除所有。

Answer 1

我决定删除模型之间共享的所有图层。 ＆＃34;分享＆＃34;实例是输入。 然后我这样做了：

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