diff options
| author | Pherkel | 2023-09-18 18:11:33 +0200 |
|---|---|---|
| committer | Pherkel | 2023-09-18 18:11:33 +0200 |
| commit | d44cf7b1cab683a8aa3876619c82226f4e6d6f3b (patch) | |
| tree | dc5a0567d5ff939320c9737e1d66e9e83f0f534c /swr2_asr/utils | |
| parent | c09ff76ba6f4c5dd5de64a401efcd27449150aec (diff) | |
fix
Diffstat (limited to 'swr2_asr/utils')
| -rw-r--r-- | swr2_asr/utils/decoder.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/swr2_asr/utils/decoder.py b/swr2_asr/utils/decoder.py index 6ffdef2..1fd002a 100644 --- a/swr2_asr/utils/decoder.py +++ b/swr2_asr/utils/decoder.py @@ -98,7 +98,8 @@ class GreedyDecoder: if greedy_type == "inference": res = self.inference(output) - res = [[DecoderOutput(words=res)]] + res = [x.split(" ") for x in res] + res = [[DecoderOutput(x)] for x in res] return res def train(self, output, labels, label_lengths): |