from local

This commit is contained in:
dohe0342 2023-04-26 12:12:26 +09:00
parent 1654d32b7f
commit b23434a2cd
2 changed files with 5 additions and 2 deletions

View File

@ -493,9 +493,12 @@ class Data2VecAudioModel(BaseFairseqModel):
print(features.size())
print(padding_mask.size())
print(padding_mask[0])
conv_feat_all = None
conv_feat_all = torch.tensor([])
for i in range(padding_mask.size()[0]):
print(padding_mask[i].nonzero())
nonzero = padding_mask[i].nonzero()
conv_feat_all =
#print(padding_mask[i].nonzero())
print(padding_mask.nonzero())
print(padding_mask.nonzero().size())