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

Speed up yesno training to finish in ~10s on CPU (#1215)

Browse Source
This commit is contained in:
Piotr Żelasko 2023-08-12 21:50:59 -04:00 committed by GitHub
parent dfccadc6b6
commit b0e8a40c89
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/yesno/ASR/tdnn/asr_datamodule.py
View File

@ -209,7 +209,7 @@ class YesNoAsrDataModule(DataModule):
sampler=train_sampler, sampler=train_sampler,
batch_size=None, batch_size=None,
num_workers=self.args.num_workers, num_workers=self.args.num_workers,
persistent_workers=False, persistent_workers=True,
) )
return train_dl return train_dl
@ -236,6 +236,7 @@ class YesNoAsrDataModule(DataModule):
batch_size=None, batch_size=None,
sampler=sampler, sampler=sampler,
num_workers=self.args.num_workers, num_workers=self.args.num_workers,
persistent_workers=True,
) )
return test_dl return test_dl