mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-09 18:12:19 +00:00
fix bug
This commit is contained in:
parent
73927ab8e8
commit
946d6ea00b
@ -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
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user