From af3ad39762575532eed60a33a67c079299188455 Mon Sep 17 00:00:00 2001 From: Holger Roth Date: Wed, 15 Jun 2022 15:07:42 -0400 Subject: [PATCH] tune number workers for challenge system --- .../breast_density_challenge/code/pt/learners/mammo_learner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 index eca65f5553..19a312ca80 100644 --- 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,