Start Multi-OS CI (#172)

2023-01-10 08:23:32 +01:00
parent 202f7bffa3
commit f80ff923e7
5 changed files with 26 additions and 25 deletions
--- a/ultralytics/yolo/v8/classify/train.py
+++ b/ultralytics/yolo/v8/classify/train.py
@ -72,7 +72,8 @@ class ClassificationTrainer(BaseTrainer):
                                               imgsz=self.args.imgsz,
                                               batch_size=batch_size if mode == "train" else (batch_size * 2),
                                               augment=mode == "train",
-                                               rank=rank)
+                                               rank=rank,
+                                               workers=self.args.workers)

    def preprocess_batch(self, batch):
        batch["img"] = batch["img"].to(self.device)