| | |
|---|
| | | speech_lengths.sum().item() * self.frontend.frame_shift * self.frontend.lfr_n / 1000 |
|---|
| | | ) |
|---|
| | | |
|---|
| | | speech.to(device=device), speech_lengths.to(device=device) |
|---|
| | | batch = {"input": speech, "input_len": speech_lengths, "key": key_batch} |
|---|
| | | if kwargs.get("return_pt", True): |
|---|
| | | speech, speech_lengths = speech.to(device=device), speech_lengths.to(device=device) |
|---|
| | | else: |
|---|
| | | speech, speech_lengths = speech.numpy(), speech_lengths.numpy() |
|---|
| | | batch = { |
|---|
| | | "input": speech, |
|---|
| | | "input_len": speech_lengths, |
|---|
| | | "key": key_batch, |
|---|
| | | data_type: "fbank", |
|---|
| | | } |
|---|
| | | result_list.append(batch) |
|---|
| | | |
|---|
| | | pbar.update(1) |
|---|
