FUNASR训练
blame | 历史 | 补丁 | 提交 | 提交对比 | ignore whitespace
游雁
2023-11-24 afae93b43dcd6f9061dec5934b6536981c7ef363 
 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)