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