在Python循环中创建模型后清理内存

我在Python中运行一个for循环,每次循环都需要在不同的数据上创建一个模型(如下所示的摘录)。每次创建的模型并未从内存中清除,导致每次循环的速度变慢。

import gcfor s in range(0, 5):  X, Y = get_data()  m = make_dgp_model(X, Y, Z_100, L)  del m  gc.collect()  print('memory: {}'.format(resource.getrusage(resource.RUSAGE_SELF).ru_maxrss / 1000000))

输出如下:

memory: 460.025856memory: 470.310912memory: 486.764544memory: 493.457408memory: 499.523584

我了解到Python使用指针引用进行内存管理,并且当变量被重用时,它不会重写内存中的内容。因此,我尝试使用del m并随后使用垃圾回收器,但似乎没有效果。我做错了什么吗?有没有办法在循环结束后完全删除m中存储的内容?


回答:

GPflow的自述文件页面有一个链接,指向一个很棒的技巧和窍门笔记本。在那里,你可以在第一项中找到你问题的答案 🙂

https://github.com/GPflow/GPflow/blob/develop/doc/source/notebooks/tips_and_tricks.ipynb

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注