We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents e494956 + f5ef58c commit bb7a566Copy full SHA for bb7a566
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