| | |
|---|
| | | if param_dict is not None: |
|---|
| | | hotword_list_or_file = param_dict.get('hotword') |
|---|
| | | export_mode = param_dict.get("export_mode", False) |
|---|
| | | clas_scale = param_dict.get('clas_scale', 1.0) |
|---|
| | | else: |
|---|
| | | hotword_list_or_file = None |
|---|
| | | clas_scale = param_dict.get('clas_scale', 1.0) |
|---|
| | | clas_scale = 1.0 |
|---|
| | | |
|---|
| | | if kwargs.get("device", None) == "cpu": |
|---|
| | | ngpu = 0 |
|---|
| | |
|---|
| | | batch_size_token_ms = batch_size_token*60 |
|---|
| | | if speech2text.device == "cpu": |
|---|
| | | batch_size_token_ms = 0 |
|---|
| | | batch_size_token_ms = max(batch_size_token_ms, sorted_data[0][0][1] - sorted_data[0][0][0]) |
|---|
| | | if len(sorted_data) > 0 and len(sorted_data[0]) > 0: |
|---|
| | | batch_size_token_ms = max(batch_size_token_ms, sorted_data[0][0][1] - sorted_data[0][0][0]) |
|---|
| | | |
|---|
| | | batch_size_token_ms_cum = 0 |
|---|
| | | beg_idx = 0 |
|---|
