From 309f5cafc12fa0af56bc912ed5570d8f3e9a5d46 Mon Sep 17 00:00:00 2001 From: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com> Date: Fri, 17 Jan 2025 19:18:49 +0800 Subject: [PATCH] Fix dataloader cleanup error with no workers (#18732) Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com> --- ultralytics/data/build.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/ultralytics/data/build.py b/ultralytics/data/build.py index 190c6ec9..33b31ba4 100644 --- a/ultralytics/data/build.py +++ b/ultralytics/data/build.py @@ -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): """