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)
|
||||
try: padding_mask = torch.cat([prompt_padding_mask, padding_mask], dim=1)
|
||||
except:
|
||||
print(padding_mask)
|
||||
print(prompt_padding_mask.size())
|
||||
print(padding_mask.size())
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user