lzr265946
2023-02-09 44fe2e811fed3d1d2341e54fa72a06931b011da1
fix bug in predictor tail_process_fn
1个文件已修改
3 ■■■■ 已修改文件
funasr/models/predictor/cif.py 3 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/models/predictor/cif.py
@@ -598,7 +598,8 @@
            mask_2 = torch.cat([ones_t, mask], dim=1)
            mask = mask_2 - mask_1
            tail_threshold = mask * tail_threshold
            alphas = torch.cat([alphas, tail_threshold], dim=1)
            alphas = torch.cat([alphas, zeros_t], dim=1)
            alphas = torch.add(alphas, tail_threshold)
        else:
            tail_threshold = torch.tensor([tail_threshold], dtype=alphas.dtype).to(alphas.device)
            tail_threshold = torch.reshape(tail_threshold, (1, 1))