FUNASR训练
parent: e451eb79 | 补丁 | 提交 | ignore whitespace
shixian.shi
2024-03-06 5b2218076bae0e22dedffa5bac7c40755b1a1589 
bug fix
1个文件已修改
2 ■■■■■ 已修改文件
funasr/models/ct_transformer/model.py 2 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 
 funasr/models/ct_transformer/model.py


@@ -359,7 +359,7 @@


                        ind_append = len_tokens - i - 1


                        for _ in range(num_append):


                            new_punc_array.insert(ind_append, 1)


        punc_array = torch.tensor(new_punc_array)


            punc_array = torch.tensor(new_punc_array)


        


        result_i = {"key": key[0], "text": new_mini_sentence_out, "punc_array": punc_array}


        results.append(result_i)