tune number workers for challenge system (#756)

holgerroth · web-flow · commit 43cdabdbb985 · 2022-06-15T20:12:08.000+01:00
diff --git a/federated_learning/breast_density_challenge/code/pt/learners/mammo_learner.py b/federated_learning/breast_density_challenge/code/pt/learners/mammo_learner.py
@@ -238,7 +238,7 @@ def initialize(self, parts: dict, fl_ctx: FLContext):
             base_dir=self.dataset_root,
         )
 
-        num_workers = 1
+        num_workers = 4  # tuned for challenge system. Please do not change.
         cache_rate = 1.0
         self.train_dataset = CacheDataset(
             data=train_datalist,

Original file line number	Diff line number	Diff line change
`@@ -238,7 +238,7 @@ def initialize(self, parts: dict, fl_ctx: FLContext):`
`238`	`238`	`base_dir=self.dataset_root,`
`239`	`239`	`)`
`240`	`240`
`241`		`- num_workers = 1`
	`241`	`+ num_workers = 4 # tuned for challenge system. Please do not change.`
`242`	`242`	`cache_rate = 1.0`
`243`	`243`	`self.train_dataset = CacheDataset(`
`244`	`244`	`data=train_datalist,`