from local

This commit is contained in:
dohe0342 2023-04-11 14:58:55 +09:00
parent d0af8e2d23
commit d01264cb9b
2 changed files with 4 additions and 4 deletions

View File

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