from local

This commit is contained in:
dohe0342 2023-04-11 14:57:17 +09:00
parent 31d2283773
commit 4e29029b17
2 changed files with 1 additions and 1 deletions

View File

@ -495,7 +495,7 @@ class Data2VecAudioModel(BaseFairseqModel):
padding_mask = None
print(padding_mask.size())
print(len((padding_mask == True).nonzero(as_tuple=True)))
print(len((padding_mask == True).nonzero(as_tuple=True)[0]))
exit()
if self.post_extract_proj is not None: