from local

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

View File

@ -493,6 +493,7 @@ class Data2VecAudioModel(BaseFairseqModel):
print(features.size())
print(padding_mask.size())
print(padding_mask.nonzero())
print(padding_mask.nonezero().size())
exit()
prompt = prompt.expand((features.size()[0], prompt.size()[0], prompt.size()[1]))
features = torch.cat([prompt, features], dim=1)