fix conformer pytorch

priyakasimbeg · priyakasimbeg · commit 21a196aab465 · 2025-08-18T19:54:36.000Z
diff --git a/algoperf/workloads/librispeech_conformer/librispeech_pytorch/workload.py b/algoperf/workloads/librispeech_conformer/librispeech_pytorch/workload.py
@@ -189,8 +189,6 @@ def _build_input_queue(
     dataloader = data_utils.cycle(
       dataloader, custom_sampler=USE_PYTORCH_DDP, use_mixup=False
     )
-
-    dataloader = map(data_utils.shard, dataloader)
     return dataloader
 
   # Does NOT apply regularization, which is left to the submitter to do in

Original file line number	Diff line number	Diff line change
`@@ -189,8 +189,6 @@ def _build_input_queue(`
`189`	`189`	`dataloader = data_utils.cycle(`
`190`	`190`	`dataloader, custom_sampler=USE_PYTORCH_DDP, use_mixup=False`
`191`	`191`	`)`
`192`		`-`
`193`		`- dataloader = map(data_utils.shard, dataloader)`
`194`	`192`	`return dataloader`
`195`	`193`
`196`	`194`	`# Does NOT apply regularization, which is left to the submitter to do in`