update

speech_asr

2023-04-17 9f90bad3f58c86143e630a9d11d8434adaa62904

1
2
3
4
5
6
7
8
9
10
11
from funasr.datasets.large_datasets.build_dataloader import LargeDataLoader
 
 
def build_dataloader(args):
    if args.dataset_type == "small":
        pass
    elif args.dataset_type == "large":
        train_iter_factory = LargeDataLoader(args, mode="train")
        valid_iter_factory = LargeDataLoader(args,  mode="valid")
    else:
        raise ValueError(f"Not supported dataset_type={args.dataset_type}")