From c5c5c55b90fb83d23b26f47219d06f2ea7d27a84 Mon Sep 17 00:00:00 2001
From: 仁迷 <haoneng.lhn@alibaba-inc.com>
Date: 星期四, 09 二月 2023 15:14:13 +0800
Subject: [PATCH] fix uniasr training bug
---
funasr/tasks/abs_task.py | 2 ++
1 files changed, 2 insertions(+), 0 deletions(-)
diff --git a/funasr/tasks/abs_task.py b/funasr/tasks/abs_task.py
index 1c8e640..7899400 100644
--- a/funasr/tasks/abs_task.py
+++ b/funasr/tasks/abs_task.py
@@ -1787,6 +1787,7 @@
collate_fn,
key_file: str = None,
batch_size: int = 1,
+ fs: dict = None,
dtype: str = np.float32,
num_workers: int = 1,
allow_variable_data_keys: bool = False,
@@ -1804,6 +1805,7 @@
dataset = IterableESPnetDataset(
data_path_and_name_and_type,
float_dtype=dtype,
+ fs=fs,
preprocess=preprocess_fn,
key_file=key_file,
)
--
Gitblit v1.9.1