We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 6a3ff16 + bb7a566 commit 576d5e3Copy full SHA for 576d5e3
algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/workload.py
@@ -161,7 +161,7 @@ def _build_input_queue(
161
batch_size=global_batch_size,
162
shuffle=train,
163
sampler=None,
164
- num_workers=4 if train else self.eval_num_workers,
+ num_workers=4,
165
prefetch_factor=10,
166
pin_memory=False,
167
drop_last=train,
algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/workload.py
@@ -187,7 +187,7 @@ def _build_input_queue(
187
batch_size=ds_iter_batch_size,
188
shuffle=not USE_PYTORCH_DDP and is_train,
189
sampler=sampler,
190
- num_workers=4 if is_train else self.eval_num_workers,
191
pin_memory=True,
192
drop_last=is_train)
193
0 commit comments