diff --git a/egs/librispeech/ASR/conformer_ctc/gigaspeech_datamodule.py b/egs/librispeech/ASR/conformer_ctc/gigaspeech_datamodule.py index 653f82873..774a61941 100644 --- a/egs/librispeech/ASR/conformer_ctc/gigaspeech_datamodule.py +++ b/egs/librispeech/ASR/conformer_ctc/gigaspeech_datamodule.py @@ -375,7 +375,8 @@ class GigaSpeechAsrDataModule(DataModule): def __iter__(self): # work horse for inexhuastible_train_dataloaders while True: - # self.total_train_cuts / self.consumed_cuts should be contained by child class + # 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: self.train_dl = self.train_dataloaders() self.consumed_cuts = 0