Skip to content

Commit 576d5e3

Browse files
Merge pull request #737 from mlcommons/dev
Dev -> main
2 parents 6a3ff16 + bb7a566 commit 576d5e3

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)