Set workers=0 for MPS Train and Val modes (#4697)

2023-09-02 04:25:03 +02:00 · 2023-09-02 04:25:03 +02:00 · a4fabfdacf
commit a4fabfdacf
parent 2bc6e647c7
14 changed files with 88 additions and 74 deletions
--- a/ultralytics/engine/validator.py
+++ b/ultralytics/engine/validator.py
@ -144,7 +144,7 @@ class BaseValidator:
            else:
                raise FileNotFoundError(emojis(f"Dataset '{self.args.data}' for task={self.args.task} not found ❌"))

-            if self.device.type == 'cpu':
+            if self.device.type in ('cpu', 'mps'):
                self.args.workers = 0  # faster CPU val as time dominated by inference, not dataloading
            if not pt:
                self.args.rect = False