Making chunksize smaller

ancestor-mithril · Aug 7, 2024 · ca18772 · ca18772
1 parent 5623b80
commit ca18772
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/dice_score_3d/metrics.py b/dice_score_3d/metrics.py
@@ -136,9 +136,7 @@ def execute_evaluate_predictions(gt_files: List[str], pred_files: List[str], reo
         ret = [evaluate_prediction(gt, pred, reorient, dtype, indices) for gt, pred in tqdm(
             list(zip(gt_files, pred_files)))]
     else:
-        chunksize = len(gt_files) // num_workers // 4  # arbitrarily chosen
-        if chunksize < 4:
-            chunksize = 1
+        chunksize = min(len(gt_files) // 50 // num_workers, 1)  # 50 is arbitrarily chosen
         ret = process_map(evaluate_prediction_wrapper,
                           [(gt, pred, reorient, dtype, indices) for gt, pred in zip(gt_files, pred_files)],
                           max_workers=num_workers, chunksize=chunksize)