mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 01:52:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Minor fixes.

Browse Source
This commit is contained in:
Fangjun Kuang 2021-08-21 08:25:34 +08:00
parent 8a8bf67faf
commit ed16585c58
1 changed files with 1 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
egs/librispeech/ASR/tdnn_lstm_ctc/asr_datamodule.py
View File

@ -137,15 +137,6 @@ class LibriSpeechAsrDataModule(DataModule):
"collect the batches.", "collect the batches.",
) )
group.add_argument(
"--num-workers-inner",
type=int,
default=8,
help="The number of sub-workers (replicated for each of "
"training dataloader workers) that parallelize "
"the I/O to collect each batch.",
)
def train_dataloaders(self) -> DataLoader: def train_dataloaders(self) -> DataLoader:
logging.info("About to get train cuts") logging.info("About to get train cuts")
cuts_train = self.train_cuts() cuts_train = self.train_cuts()
@ -227,7 +218,7 @@ class LibriSpeechAsrDataModule(DataModule):
train, train,
sampler=train_sampler, sampler=train_sampler,
batch_size=None, batch_size=None,
num_workers=2, num_workers=self.args.num_workers,
persistent_workers=False, persistent_workers=False,
) )