Merge pull request #737 from mlcommons/dev

Dev -> main
mlcommons · Mar 27, 2024 · 576d5e3 · 576d5e3
2 parents 6a3ff16 + bb7a566
commit 576d5e3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/workload.py b/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/workload.py
@@ -161,7 +161,7 @@ def _build_input_queue(
             batch_size=global_batch_size,
             shuffle=train,
             sampler=None,
-            num_workers=4 if train else self.eval_num_workers,
+            num_workers=4,
             prefetch_factor=10,
             pin_memory=False,
             drop_last=train,

diff --git a/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/workload.py b/algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/workload.py
@@ -187,7 +187,7 @@ def _build_input_queue(
         batch_size=ds_iter_batch_size,
         shuffle=not USE_PYTORCH_DDP and is_train,
         sampler=sampler,
-        num_workers=4 if is_train else self.eval_num_workers,
+        num_workers=4,
         pin_memory=True,
         drop_last=is_train)