FUNASR训练
parent: c3442d95 | 补丁 | 提交 | ignore whitespace
shixian.shi
2024-01-12 0c75e62c6e06e056ca9671bd6f8c9b0c5738fd51 
update device bug
1个文件已修改
2 ■■■■■ 已修改文件
funasr/bin/inference.py 2 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 
 funasr/bin/inference.py


@@ -405,7 +405,7 @@


                spk_embedding = result['spk_embedding']


                labels = self.cb_model(spk_embedding, oracle_num=self.preset_spk_num)


                del result['spk_embedding']


                sv_output = postprocess(all_segments, None, labels, spk_embedding)


                sv_output = postprocess(all_segments, None, labels, spk_embedding.cpu())


                if self.spk_mode == 'vad_segment':


                    sentence_list = []


                    for res, vadsegment in zip(restored_data, vadsegments):