Skip to content

Commit e777fd2

Browse files
committed
dyad: update multiprocessing approach for DLIO
This commit tweaks the DLIO config file to use forking for multiprocessing instead of spawning
1 parent e0d6589 commit e777fd2

File tree

1 file changed

+2
-2
lines changed
  • 2024-HPCIC-AWS/JupyterNotebook/tutorial/supplementary/dyad/dlio_extensions/configs/workload

1 file changed

+2
-2
lines changed

2024-HPCIC-AWS/JupyterNotebook/tutorial/supplementary/dyad/dlio_extensions/configs/workload/dyad_unet3d_demo.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ reader:
2020
read_threads: 1
2121
file_shuffle: seed
2222
sample_shuffle: seed
23-
multiprocessing_context: spawn
23+
multiprocessing_context: fork
2424
data_loader_classname: dyad_torch_data_loader.DyadTorchDataLoader
2525
data_loader_sampler: index
2626

@@ -32,4 +32,4 @@ checkpoint:
3232
checkpoint_folder: checkpoints/unet3d
3333
checkpoint_after_epoch: 5
3434
epochs_between_checkpoints: 2
35-
model_size: 499153191
35+
model_size: 499153191

0 commit comments

Comments
 (0)