HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Start Multi-OS CI (#172)

Browse source
This commit is contained in:
Glenn Jocher 2023-01-10 08:23:32 +01:00 • committed by GitHub
parent 202f7bffa3
commit f80ff923e7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 26 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

3
ultralytics/yolo/v8/classify/train.py
View file

@ -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)

5
ultralytics/yolo/v8/classify/val.py
View file

@ -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