chief: &gpu mem: 1.4Ti cpu: 24 num_accelerators: 16 accelerator_type: a100 dataset_dispatcher: mem: 2Gi cpu: 2 num_dataset_workers: 4 dataset_worker: mem: 14Gi cpu: 2