This commit is contained in:
Guo Liyong 2021-11-03 18:38:08 +08:00
parent 73927ab8e8
commit 946d6ea00b

View File

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