FUNASR训练
parent: 5c28b4d6 | 补丁 | 提交 | ignore whitespace
pointerhacker
2024-10-15 70645e48072bf193fbf069949f1d2b10fddac8a3 
数据并行可能导致的模型训练报错 (#2139)

* fix: 修复数据并行训练中ä¼可能会出现的错误

* fix: 修复数据并行训练中ä¼可能会出现的错误

* fix: 修复数据并行ènot need tensor

---------

Co-authored-by: zhaochaojin <zhaochaojin@didiglobal.com>
1个文件已修改
3 ■■■■■ 已修改文件
funasr/models/sense_voice/model.py 3 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 
 funasr/models/sense_voice/model.py


@@ -555,7 +555,8 @@


        ilens: torch.Tensor,


    ):


        """Embed positions in tensor."""


        masks = sequence_mask(ilens, device=ilens.device)[:, None, :]


        maxlen = xs_pad.shape[1]


        masks = sequence_mask(ilens, maxlen = maxlen, device=ilens.device)[:, None, :]




        xs_pad *= self.output_size() ** 0.5