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
fd9690653e
commit
a9008aa082
Binary file not shown.
@ -496,6 +496,7 @@ class Data2VecAudioModel(BaseFairseqModel):
|
|||||||
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)
|
||||||
try: padding_mask = torch.cat([prompt_padding_mask, padding_mask], dim=1)
|
try: padding_mask = torch.cat([prompt_padding_mask, padding_mask], dim=1)
|
||||||
except:
|
except:
|
||||||
|
print(padding_mask)
|
||||||
print(prompt_padding_mask.size())
|
print(prompt_padding_mask.size())
|
||||||
print(padding_mask.size())
|
print(padding_mask.size())
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user