mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

from local

Browse Source
This commit is contained in:
dohe0342 2022-12-22 14:44:42 +09:00
parent 4250de81a4
commit 9bba4015cb
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/.data2vec_audio.py.swp
View File

Binary file not shown.

3
egs/librispeech/ASR/pruned_transducer_stateless_d2v_v2/data2vec_audio.py
View File

@ -253,7 +253,8 @@ class Data2VecAudioModel(BaseFairseqModel):
mask_dropout=self.cfg.mask_dropout, mask_dropout=self.cfg.mask_dropout,
) )
mask_indices = torch.from_numpy(mask_indices).to(x.device) mask_indices = torch.from_numpy(mask_indices).to(x.device)
x = index_put(x, mask_indices, self.mask_emb) #x = index_put(x, mask_indices, self.mask_emb)
x = index_put(x, mask_indices, 0)
else: else:
mask_indices = None mask_indices = None