from local
This commit is contained in:
parent
d01264cb9b
commit
f56ad643e6
Binary file not shown.
@ -496,9 +496,9 @@ class Data2VecAudioModel(BaseFairseqModel):
|
|||||||
|
|
||||||
print(padding_mask.size())
|
print(padding_mask.size())
|
||||||
print((padding_mask[0] == True).nonzero(as_tuple=True)[0])
|
print((padding_mask[0] == True).nonzero(as_tuple=True)[0])
|
||||||
print((padding_mask[1] == True).nonzero(as_tuple=True)[0])
|
print((padding_mask[1] == True).nonzero(as_tuple=True)[0][1])
|
||||||
print((padding_mask[2] == True).nonzero(as_tuple=True)[0])
|
print((padding_mask[2] == True).nonzero(as_tuple=True)[0][2])
|
||||||
print((padding_mask[3] == True).nonzero(as_tuple=True)[0])
|
print((padding_mask[3] == True).nonzero(as_tuple=True)[0][3])
|
||||||
exit()
|
exit()
|
||||||
|
|
||||||
if self.post_extract_proj is not None:
|
if self.post_extract_proj is not None:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user