from local

This commit is contained in:
dohe0342 2023-04-11 14:51:40 +09:00
parent f41a2ee2a1
commit 114cc475a7
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].index(True))
print((padding_mask[9] == True).nonzero(as_tuple=True)[0])
exit()
if self.post_extract_proj is not None: