From b0e8a40c8932d82d356b8a2ad4948331eae9749e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Piotr=20=C5=BBelasko?= <petezor@gmail.com>
Date: Sat, 12 Aug 2023 21:50:59 -0400
Subject: [PATCH] Speed up yesno training to finish in ~10s on CPU (#1215)

---
 egs/yesno/ASR/tdnn/asr_datamodule.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/egs/yesno/ASR/tdnn/asr_datamodule.py b/egs/yesno/ASR/tdnn/asr_datamodule.py
index 3c1682fa1..ada8c1a6c 100644
--- a/egs/yesno/ASR/tdnn/asr_datamodule.py
+++ b/egs/yesno/ASR/tdnn/asr_datamodule.py
@@ -209,7 +209,7 @@ class YesNoAsrDataModule(DataModule):
             sampler=train_sampler,
             batch_size=None,
             num_workers=self.args.num_workers,
-            persistent_workers=False,
+            persistent_workers=True,
         )
 
         return train_dl
@@ -236,6 +236,7 @@ class YesNoAsrDataModule(DataModule):
             batch_size=None,
             sampler=sampler,
             num_workers=self.args.num_workers,
+            persistent_workers=True,
         )
         return test_dl