from local

This commit is contained in:
dohe0342 2023-04-11 14:54:03 +09:00
parent f00295c4fb
commit 19f14e3242
2 changed files with 1 additions and 1 deletions

View File

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