Skip to content

Commit 28f6253

Browse files
committed
fix DataLoader memory leak. test=develop
1 parent eae3405 commit 28f6253

File tree

1 file changed

+1
-15
lines changed

1 file changed

+1
-15
lines changed

python/paddle/fluid/dataloader/dataloader_iter.py

Lines changed: 1 addition & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -125,13 +125,6 @@ def __init__(self, loader):
125125

126126
self._init_thread()
127127

128-
# if user exit python program when dataloader is still
129-
# iterating, resource may no release safely, so we
130-
# add __del__ function to to CleanupFuncRegistrar
131-
# to make sure __del__ is always called when program
132-
# exit for resoure releasing safely
133-
CleanupFuncRegistrar.register(self.__del__)
134-
135128
def _init_thread(self):
136129
self._var_names = [v.name for v in self._feed_list]
137130
self._shapes = [v.shape for v in self._feed_list]
@@ -231,7 +224,7 @@ def _shutdown_thread(self):
231224
self._thread_done_event.set()
232225
if self._thread is not threading.current_thread():
233226
self._thread.join()
234-
self._thread = None
227+
self._thread = None
235228

236229
# python2 compatibility
237230
def next(self):
@@ -287,13 +280,6 @@ def __init__(self, loader):
287280
self._init_thread()
288281
self._shutdown = False
289282

290-
# if user exit python program when dataloader is still
291-
# iterating, resource may no release safely, so we
292-
# add __del__ function to to CleanupFuncRegistrar
293-
# to make sure __del__ is always called when program
294-
# exit for resoure releasing safely
295-
CleanupFuncRegistrar.register(self.__del__)
296-
297283
def _init_workers(self):
298284
# multiprocess worker and indice queue list initial as empty
299285
self._workers = []

0 commit comments

Comments
 (0)