from local

This commit is contained in:
dohe0342 2023-04-26 12:07:39 +09:00
parent b402fcf667
commit 1dc02ddb19
2 changed files with 2 additions and 0 deletions

View File

@ -493,6 +493,8 @@ class Data2VecAudioModel(BaseFairseqModel):
print(features.size())
print(padding_mask.size())
print(padding_mask[0])
for i in range(padding_mask.size()[0]):
print(padding_mask[i].nonzero())
print(padding_mask.nonzero())
print(padding_mask.nonzero().size())