From a2430d047897b4aad3c4cdf841c72343cbc8d5bb Mon Sep 17 00:00:00 2001
From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Date: Tue, 21 Jan 2025 00:12:13 +0800
Subject: [PATCH] Catch and ignore exception in dataloader cleanup (#18772)

Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
---
 ultralytics/data/build.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/ultralytics/data/build.py b/ultralytics/data/build.py
index 33b31ba4..46823830 100644
--- a/ultralytics/data/build.py
+++ b/ultralytics/data/build.py
@@ -49,11 +49,15 @@ class InfiniteDataLoader(dataloader.DataLoader):
 
     def __del__(self):
         """Ensure that workers are terminated."""
-        if hasattr(self.iterator, "_workers"):
+        try:
+            if not hasattr(self.iterator, "_workers"):
+                return
             for w in self.iterator._workers:  # force terminate
                 if w.is_alive():
                     w.terminate()
             self.iterator._shutdown_workers()  # cleanup
+        except Exception:
+            pass
 
     def reset(self):
         """