update

speech_asr

2023-04-18 05d4176e88bdde0ce615cb22daf7af725a496020

 funasr/utils/build_dataloader.py

@@ -1,9 +1,10 @@
from funasr.datasets.large_datasets.build_dataloader import LargeDataLoader

from funasr.datasets.small_datasets.build_dataloader import build_dataloader

def build_dataloader(args):
    if args.dataset_type == "small":
        pass
        train_iter_factory = LargeDataLoader(args, mode="train")
        valid_iter_factory = LargeDataLoader(args, mode="valid")
    elif args.dataset_type == "large":
        train_iter_factory = LargeDataLoader(args, mode="train")
        valid_iter_factory = LargeDataLoader(args,  mode="valid")