HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Fix dataloader cleanup error with no workers (#18732)

Browse source 
Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
This commit is contained in:
Mohammed Yasin 2025-01-17 19:18:49 +08:00 • committed by GitHub
parent b1042c4da8
commit 309f5cafc1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

9
ultralytics/data/build.py
View file

@ -49,10 +49,11 @@ class InfiniteDataLoader(dataloader.DataLoader):
def __del__(self):
"""Ensure that workers are terminated."""
for w in self.iterator._workers: # force terminate
if w.is_alive():
w.terminate()
self.iterator._shutdown_workers() # cleanup
if hasattr(self.iterator, "_workers"):
for w in self.iterator._workers: # force terminate
if w.is_alive():
w.terminate()
self.iterator._shutdown_workers() # cleanup
def reset(self):
"""