From 28cb2f2213e08e0767c8142ec7b1b31d2ded306f Mon Sep 17 00:00:00 2001 From: Andreas Moldskred Date: Fri, 3 May 2024 14:06:03 -0700 Subject: [PATCH] `ultralytics 8.2.8` fix `os.sched_setaffinity()` AttributeError (#11232) Co-authored-by: UltralyticsAssistant Co-authored-by: Glenn Jocher --- ultralytics/__init__.py | 2 +- ultralytics/data/build.py | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index bb2ea3f6..4c98c5c0 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = "8.2.7" +__version__ = "8.2.8" from ultralytics.data.explorer.explorer import Explorer from ultralytics.models import RTDETR, SAM, YOLO, YOLOWorld diff --git a/ultralytics/data/build.py b/ultralytics/data/build.py index d0773737..6fa611cc 100644 --- a/ultralytics/data/build.py +++ b/ultralytics/data/build.py @@ -20,7 +20,7 @@ from ultralytics.data.loaders import ( autocast_list, ) from ultralytics.data.utils import IMG_FORMATS, VID_FORMATS -from ultralytics.utils import NUM_THREADS, RANK, colorstr +from ultralytics.utils import LINUX, NUM_THREADS, RANK, colorstr from ultralytics.utils.checks import check_file from .dataset import GroundingDataset, YOLODataset, YOLOMultiModalDataset @@ -81,7 +81,8 @@ def seed_worker(worker_id): # noqa worker_seed = torch.initial_seed() % 2**32 np.random.seed(worker_seed) random.seed(worker_seed) - os.sched_setaffinity(0, range(NUM_THREADS)) # fix https://github.com/ultralytics/ultralytics/pull/11195 + if LINUX and hasattr(os, "sched_setaffinity"): # unsupported on macOS and Windows + os.sched_setaffinity(0, range(NUM_THREADS)) # fix https://github.com/ultralytics/ultralytics/pull/11195 def build_yolo_dataset(cfg, img_path, batch, data, mode="train", rect=False, stride=32, multi_modal=False):