UltralyticsAssistant 3 months ago
parent d56a689309
commit 20de6d2690
  1. 4
      ultralytics/engine/validator.py

@ -147,7 +147,9 @@ class BaseValidator:
elif self.args.task == "classify":
self.data = check_cls_dataset(self.args.data, split=self.args.split)
else:
raise FileNotFoundError(emojis(f"Dataset '{self.args.data}' for task={self.args.task} not found ❌"))
raise FileNotFoundError(
emojis(f"Dataset '{self.args.data}' for task={self.args.task} not found ❌")
)
if self.device.type in {"cpu", "mps"}:
self.args.workers = 0 # faster CPU val as time dominated by inference, not dataloading

Loading…
Cancel
Save