Skip to content

Commit bb7a566

Browse files
Merge pull request #736 from mlcommons/speech_data_hot_fix
set num_workers for librispeech back to 4
2 parents e494956 + f5ef58c commit bb7a566

File tree

2 files changed

+2
-2
lines changed
  • algorithmic_efficiency/workloads/librispeech_conformer

2 files changed

+2
-2
lines changed

algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/workload.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -161,7 +161,7 @@ def _build_input_queue(
161161
batch_size=global_batch_size,
162162
shuffle=train,
163163
sampler=None,
164-
num_workers=4 if train else self.eval_num_workers,
164+
num_workers=4,
165165
prefetch_factor=10,
166166
pin_memory=False,
167167
drop_last=train,

algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/workload.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -187,7 +187,7 @@ def _build_input_queue(
187187
batch_size=ds_iter_batch_size,
188188
shuffle=not USE_PYTORCH_DDP and is_train,
189189
sampler=sampler,
190-
num_workers=4 if is_train else self.eval_num_workers,
190+
num_workers=4,
191191
pin_memory=True,
192192
drop_last=is_train)
193193

0 commit comments

Comments
 (0)