From 05d4176e88bdde0ce615cb22daf7af725a496020 Mon Sep 17 00:00:00 2001
From: speech_asr <wangjiaming.wjm@alibaba-inc.com>
Date: 星期二, 18 四月 2023 19:28:33 +0800
Subject: [PATCH] update
---
funasr/utils/build_dataloader.py | 5 +++--
1 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/funasr/utils/build_dataloader.py b/funasr/utils/build_dataloader.py
index 59b19ba..b0836fe 100644
--- a/funasr/utils/build_dataloader.py
+++ b/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")
--
Gitblit v1.9.1