fix comments

This commit is contained in:
Guo Liyong 2021-11-03 18:28:54 +08:00
parent 6864cd6ad8
commit 73927ab8e8

View File

@ -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