FUNASR训练
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
chenmengzheAAA
2023-09-14 be9d6165b31b4243c7f0ab8741e9f7664210acdf 
 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)