reduce dataload_workers

InnerEye/ML/configs/classification/CovidHierarchicalModel.py

@@ -74,7 +74,7 @@ def __init__(self, covid_dataset_id: str = COVID_DATASET_ID, **kwargs: Any):
  non_image_feature_channels=[],
  numerical_columns=[],
  use_mixed_precision=False,
- num_dataload_workers=12,
+ num_dataload_workers=2,
  multiprocessing_start_method=MultiprocessingStartMethod.fork,
  train_batch_size=64,
  optimizer_type=OptimizerType.Adam,

InnerEye/ML/configs/segmentation/GbmBase.py

@@ -34,7 +34,7 @@ def __init__(self, **kwargs: Any) -> None:
  colours=[(255, 255, 255)] * len(fg_classes),
  fill_holes=[False] * len(fg_classes),
  roi_interpreted_types=["ORGAN"] * len(fg_classes),
- num_dataload_workers=8,
+ num_dataload_workers=2,
  mask_id=None,
  norm_method=PhotometricNormalizationMethod.MriWindow,
  trim_percentiles=(1, 99),

InnerEye/ML/configs/segmentation/Lung.py

@@ -39,7 +39,7 @@ def __init__(self, **kwargs: Any) -> None:
  fill_holes=[False] * len(fg_classes),
  roi_interpreted_types=["ORGAN"] * len(fg_classes),
  largest_connected_component_foreground_classes=["lung_r", "lung_l", "heart"],
- num_dataload_workers=8,
+ num_dataload_workers=2,
  norm_method=PhotometricNormalizationMethod.CtWindow,
  level=40,
  window=400,