funasr/datasets/large_datasets/datapipes/batch.py @@ -39,7 +39,7 @@ self.batch_mode = batch_mode def set_epoch(self, epoch): self.epoch = epoch self.datapipe.set_epoch(epoch) def __iter__(self): buffer = []
