from local

This commit is contained in:
dohe0342 2023-04-26 12:03:38 +09:00
parent 2d589f91e8
commit dc18cd3712
2 changed files with 1 additions and 1 deletions

View File

@ -493,7 +493,7 @@ class Data2VecAudioModel(BaseFairseqModel):
print(features.size())
print(padding_mask.size())
print(padding_mask.nonzero())
print(padding_mask.nonezero().size())
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)