![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
内存泄漏
Ai君臣
通俗说技术AI,一针见血。AI改变世界!立志做一款让人偷懒依赖的个人定制的管家。可信可赖可知心。有志之士互勉!wx公号一休哥
展开
-
tensorflow checkpoint模型文件和pb文件加载,这样最好,预测时间短,不会泄露内存
背景:tensorflow需要加载checkpoint文件模型或者pb格式模型。问题:怎么加载保存的checkpoint并预测,加载pb格式文件加载时多次预测会内存泄漏。解决方法1、checkpoint格式加载与预测# 加载 self.graph = tf.Graph() # 为每个类(实例)单独创建一个graph with self.graph.as_default(): self.check_point_file = tf.train.lat原创 2020-09-11 10:08:03 · 764 阅读 · 0 评论 -
记录一次内存泄漏的修复过程
1、现象通过Flask API请求调用一个实例的方法,加载pysparnn的索引,每次调用都增加800M,怀疑内存泄漏。2 、工具:memory_profiler, pymplermomery_profiler 使用:这样每次调用的话,可以知道每一行代码调用是否增加了内存。from memory_profiler import profile@profiled...原创 2020-04-28 18:37:17 · 1377 阅读 · 0 评论 -
怎么增加第三方库的方法,实现自己的需要呢,比如修复内存泄漏。
思路通过动态增加类的方法去实现自己的功能。首先:动态增加类方法。class MyObj(object): def __init__(self, val): self.val = valdef new_method(self, value): return self.val + valueobj = MyObj(3)obj.method = new...原创 2020-04-28 18:22:36 · 130 阅读 · 0 评论