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): """