Skip to content

Commit

Permalink
Merge pull request espnet#5880 from wanchichen/ez_patch
Browse files Browse the repository at this point in the history
Fix missing flag in EZ streaming iterator
  • Loading branch information
sw005320 authored Aug 28, 2024
2 parents 899dd99 + a8d2a45 commit 7b44d96
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions espnetez/task.py
Original file line number Diff line number Diff line change
Expand Up @@ -353,6 +353,7 @@ def build_streaming_iterator(
ngpu: int = 0,
inference: bool = False,
mode: Optional[str] = None,
multi_task_dataset: bool = False,
) -> DataLoader:
"""Build DataLoader using iterable dataset"""
if mode == "train" and cls.train_dataloader is not None:
Expand Down

0 comments on commit 7b44d96

Please sign in to comment.