🔀 [AAAI|Merge] branch 'main' into feature/use-preprocess-target

henrytsui000 · henrytsui000 · commit 782c233889d7 · 2025-07-14T15:30:43.000+08:00
diff --git a/yolo/aaai.py b/yolo/aaai.py
@@ -196,10 +196,10 @@ def setup(self, stage=None):
         )
 
     def train_dataloader(self):
-        return DataLoader(self.train_dataset, batch_size=self.train_batch, shuffle=self.cfg.task.data.shuffle)
+        return DataLoader(self.train_dataset, batch_size=self.train_batch, shuffle=self.cfg.task.data.shuffle, num_workers=self.cfg.cpu_num)
 
     def val_dataloader(self):
-        return DataLoader(self.val_dataset, batch_size=self.valid_batch, shuffle=False)
+        return DataLoader(self.val_dataset, batch_size=self.valid_batch, shuffle=False, num_workers=self.cfg.cpu_num)
 
 
 @hydra.main(config_path="config", config_name="aaai_config", version_base=None)

Original file line number	Diff line number	Diff line change
`@@ -196,10 +196,10 @@ def setup(self, stage=None):`
`196`	`196`	`)`
`197`	`197`
`198`	`198`	`def train_dataloader(self):`
`199`		`- return DataLoader(self.train_dataset, batch_size=self.train_batch, shuffle=self.cfg.task.data.shuffle)`
	`199`	`+ return DataLoader(self.train_dataset, batch_size=self.train_batch, shuffle=self.cfg.task.data.shuffle, num_workers=self.cfg.cpu_num)`
`200`	`200`
`201`	`201`	`def val_dataloader(self):`
`202`		`- return DataLoader(self.val_dataset, batch_size=self.valid_batch, shuffle=False)`
	`202`	`+ return DataLoader(self.val_dataset, batch_size=self.valid_batch, shuffle=False, num_workers=self.cfg.cpu_num)`
`203`	`203`
`204`	`204`
`205`	`205`	`@hydra.main(config_path="config", config_name="aaai_config", version_base=None)`