from local
This commit is contained in:
parent
d0af8e2d23
commit
d01264cb9b
Binary file not shown.
@ -495,10 +495,10 @@ class Data2VecAudioModel(BaseFairseqModel):
|
|||||||
padding_mask = None
|
padding_mask = None
|
||||||
|
|
||||||
print(padding_mask.size())
|
print(padding_mask.size())
|
||||||
print((padding_mask[0] == 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][0])
|
print((padding_mask[1] == True).nonzero(as_tuple=True)[0])
|
||||||
print((padding_mask[2] == True).nonzero(as_tuple=True)[0][0])
|
print((padding_mask[2] == True).nonzero(as_tuple=True)[0])
|
||||||
print((padding_mask[3] == True).nonzero(as_tuple=True)[0][0])
|
print((padding_mask[3] == True).nonzero(as_tuple=True)[0])
|
||||||
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