FUNASR训练
parent: c880db53 | 补丁 | 提交 | ignore whitespace
维石
2024-06-28 c3ec6b9b9e44657f5818b8888253ba22099558f9 
update cif export
1个文件已修改
2 ■■■■■ 已修改文件
funasr/models/paraformer/cif_predictor.py 2 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 
 funasr/models/paraformer/cif_predictor.py


@@ -449,7 +449,7 @@


        mask = mask.transpose(-1, -2).float()



        mask = mask.squeeze(-1)



        hidden, alphas, token_num = self.tail_process_fn(hidden, alphas, mask=mask)



        acoustic_embeds, cif_peak = cif_v1_export(hidden, alphas, self.threshold)



        acoustic_embeds, cif_peak = cif_export(hidden, alphas, self.threshold)







        return acoustic_embeds, token_num, alphas, cif_peak