from local

This commit is contained in:
dohe0342 2023-04-11 14:55:39 +09:00
parent 5d2bd3ce8a
commit cb3edc6c37
2 changed files with 2 additions and 1 deletions

View File

@ -493,7 +493,8 @@ class Data2VecAudioModel(BaseFairseqModel):
padding_mask = (1 - padding_mask.flip([-1]).cumsum(-1).flip([-1])).bool()
else:
padding_mask = None
print(padding_mask.size())
print((padding_mask == True).nonzero(as_tuple=True).size())
exit()