speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add sampling rate parameter

Browse Source
This commit is contained in:
hnluo 2023-02-05 12:10:12 +08:00 committed by GitHub
parent 16d3783a54
commit aa456c5e44
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/tasks/abs_task.py
View File

@ -1783,6 +1783,7 @@ class AbsTask(ABC):
collate_fn, collate_fn,
key_file: str = None, key_file: str = None,
batch_size: int = 1, batch_size: int = 1,
fs: dict = None,
dtype: str = np.float32, dtype: str = np.float32,
num_workers: int = 1, num_workers: int = 1,
allow_variable_data_keys: bool = False, allow_variable_data_keys: bool = False,
@ -1800,6 +1801,7 @@ class AbsTask(ABC):
dataset = IterableESPnetDataset( dataset = IterableESPnetDataset(
data_path_and_name_and_type, data_path_and_name_and_type,
float_dtype=dtype, float_dtype=dtype,
fs=fs,
preprocess=preprocess_fn, preprocess=preprocess_fn,
key_file=key_file, key_file=key_file,
) )