FUNASR训练
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
shixian.shi
2023-11-23 c9f1b4e8a2e903f74de20d019e70307c26e93c3e 
 funasr/export/models/predictor/cif.py


@@ -288,7 +288,7 @@






        fire_place = integrate >= threshold



        integrate = torch.where(fire_place,



                                integrate - torch.ones([batch_size], device=alphas.device),



                                integrate - torch.ones([batch_size], device=alphas.device)*threshold,



                                integrate)







    fires = torch.stack(list_fires, 1)