Start Multi-OS CI (#172)
This commit is contained in:
@ -131,7 +131,7 @@ def smart_request(*args, retry=3, timeout=30, thread=True, code=-1, method="post
|
||||
|
||||
|
||||
@TryExcept()
|
||||
def sync_analytics(cfg, all_keys=False, enabled=True):
|
||||
def sync_analytics(cfg, all_keys=False, enabled=False):
|
||||
"""
|
||||
Sync analytics data if enabled in the global settings
|
||||
|
||||
|
@ -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)
|
||||
|
@ -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
|
||||
|
Reference in New Issue
Block a user