仁迷
2023-02-09 c5c5c55b90fb83d23b26f47219d06f2ea7d27a84
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,
        )