R 中的 Keras:多次训练后释放内存

Ing*_*ifs 5 memory r keras tensorflow tensorflow2.0

我正在 R 中训练多个模型。一段时间后,我的内存不足了。

从基本的谷歌搜索来看,张量流会话似乎在 R 中的对象被覆盖后将内容保存在内存中。这是其他人遇到的问题,但是我没有看到任何对 R 中的 keras 有帮助的答案。

Keras:完成训练过程后释放内存

Tensorflow2.0:GPU 在超参数调整循环期间内存不足

我尝试在每个循环后运行这些命令:

rm(model)

k_clear_session()

tf$compat$v1$keras$backend$clear_session()

但这些问题依然存在。关于如何释放 Keras 使用的内存有什么想法吗?

我在笔记本电脑上运行此代码,并且我很确定我没有 GPU。

bri*_*icx 1

我记得 R 中存在一些内存问题(不记得是 keras 还是其他问题),但以下一项或多项组合应该会有所帮助:

remove(list=ls()) #remove your objects

gc() #garbage collection

.rs.restartR() #restart the R session (clean the memory but doesn't detach your packages)

Run Code Online (Sandbox Code Playgroud)