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/val.py
+++ b/ultralytics/yolo/v8/classify/val.py
@ -36,7 +36,10 @@ class ClassificationValidator(BaseValidator):
        return self.metrics.results_dict

    def get_dataloader(self, dataset_path, batch_size):
-        return build_classification_dataloader(path=dataset_path, imgsz=self.args.imgsz, batch_size=batch_size)
+        return build_classification_dataloader(path=dataset_path,
+                                               imgsz=self.args.imgsz,
+                                               batch_size=batch_size,
+                                               workers=self.args.workers)

    def print_results(self):
        pf = '%22s' + '%11.3g' * len(self.metrics.keys)  # print format