| | |
|---|
| | | ngram_weight: float = 0.9, |
|---|
| | | nbest: int = 1, |
|---|
| | | num_workers: int = 1, |
|---|
| | | param_dict: dict = None, |
|---|
| | | **kwargs, |
|---|
| | | ): |
|---|
| | | assert check_argument_types() |
|---|
| | |
|---|
| | | def _forward(data_path_and_name_and_type, |
|---|
| | | raw_inputs: Union[np.ndarray, torch.Tensor] = None, |
|---|
| | | output_dir_v2: Optional[str] = None, |
|---|
| | | fs: dict = None, |
|---|
| | | param_dict: dict = None, |
|---|
| | | **kwargs, |
|---|
| | | ): |
|---|
| | | # 3. Build data-iterator |
|---|
| | | if data_path_and_name_and_type is None and raw_inputs is not None: |
|---|
| | |
|---|
| | | data_path_and_name_and_type, |
|---|
| | | dtype=dtype, |
|---|
| | | batch_size=batch_size, |
|---|
| | | fs=fs, |
|---|
| | | mc=True, |
|---|
| | | key_file=key_file, |
|---|
| | | num_workers=num_workers, |
|---|
| | | preprocess_fn=ASRTask.build_preprocess_fn(speech2text.asr_train_args, False), |
|---|
| | |
|---|
| | | return asr_result_list |
|---|
| | | |
|---|
| | | return _forward |
|---|
| | | |
|---|
| | | def set_parameters(language: str = None, |
|---|
| | | sample_rate: Union[int, Dict[Any, int]] = None): |
|---|
| | | if language is not None: |
|---|
| | | global global_asr_language |
|---|
| | | global_asr_language = language |
|---|
| | | if sample_rate is not None: |
|---|
| | | global global_sample_rate |
|---|
| | | global_sample_rate = sample_rate |
|---|
| | | |
|---|
| | | |
|---|
| | | def get_parser(): |
|---|
| | | parser = config_argparse.ArgumentParser( |
|---|
