modelscope/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR.git synced 2026-10-12 02:19:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

数据并行可能导致的模型训练报错 (#2139)

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

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

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

---------

Co-authored-by: zhaochaojin <[email protected]>
This commit is contained in:
pointerhacker
2024-10-15 17:50:51 +08:00
committed by GitHub
co-authored by zhaochaojin
parent 5c28b4d612
commit 70645e4807
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
funasr/models/sense_voice/model.py
+2 -1
View File
@@ -555,7 +555,8 @@ class SenseVoiceEncoderSmall(nn.Module):
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