mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
from local
This commit is contained in:
parent
383bf6aef7
commit
208a473c5e
Binary file not shown.
Binary file not shown.
@ -495,7 +495,7 @@ class Data2VecAudioModel(BaseFairseqModel):
|
|||||||
features = torch.cat([prompt, features], dim=1)
|
features = torch.cat([prompt, features], dim=1)
|
||||||
prompt_padding_mask = torch.zeros(prompt.size()[0], prompt.size()[1]).type(torch.BoolTensor).to(features.device)
|
prompt_padding_mask = torch.zeros(prompt.size()[0], prompt.size()[1]).type(torch.BoolTensor).to(features.device)
|
||||||
padding_mask = torch.cat([prompt_padding_mask, padding_mask], dim=1)
|
padding_mask = torch.cat([prompt_padding_mask, padding_mask], dim=1)
|
||||||
print(padding_mask)
|
print('fuccckkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkkk!!!!!!!!!!!!!!!!!!1')
|
||||||
|
|
||||||
features = self.layer_norm(features)
|
features = self.layer_norm(features)
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user