from local

This commit is contained in:
dohe0342 2023-04-26 12:05:19 +09:00
parent dc18cd3712
commit b402fcf667
2 changed files with 2 additions and 0 deletions

View File

@ -492,8 +492,10 @@ class Data2VecAudioModel(BaseFairseqModel):
if prompt is not None:
print(features.size())
print(padding_mask.size())
print(padding_mask[0])
print(padding_mask.nonzero())
print(padding_mask.nonzero().size())
exit()
prompt = prompt.expand((features.size()[0], prompt.size()[0], prompt.size()[1]))
features = torch.cat([prompt, features], dim=1)