From 946d6ea00b8d5a3767ce8d01d7eda1354ca47901 Mon Sep 17 00:00:00 2001 From: Guo Liyong Date: Wed, 3 Nov 2021 18:38:08 +0800 Subject: [PATCH] fix bug --- egs/librispeech/ASR/conformer_ctc/gigaspeech_datamodule.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/conformer_ctc/gigaspeech_datamodule.py b/egs/librispeech/ASR/conformer_ctc/gigaspeech_datamodule.py index 774a61941..7119bee62 100644 --- a/egs/librispeech/ASR/conformer_ctc/gigaspeech_datamodule.py +++ b/egs/librispeech/ASR/conformer_ctc/gigaspeech_datamodule.py @@ -377,7 +377,7 @@ class GigaSpeechAsrDataModule(DataModule): while True: # self.total_train_cuts == 0 for the first run # self.consumed_cuts == self.total_train_cuts for recreating dataloader - if self.total_train_cuts == 0 and self.consumed_cuts == self.total_train_cuts: + if self.total_train_cuts == 0 or self.consumed_cuts == self.total_train_cuts: self.train_dl = self.train_dataloaders() self.consumed_cuts = 0